marius311 / muse-implicit-paper

Improved Marginal Unbiased Score Expansion (MUSE) via Implicit Differentiation

Code to produce plots from http://arxiv.org/abs/2209.10512

Usage:

git clone https://github.com/marius311/muse-implicit-paper.git
cd muse-implicit-paper
poetry install

Then open JupyterLab and run the notebooks, making sure to you have the Poetry kernel (pip install poetry-kernel) installed in the Python environment that runs your JupyterLab and selected for the notebooks.