.gitignore
.readthedocs.yaml
LICENSE
README.md
pyproject.toml
requirements.txt
setup.py
.github/workflows/publish-to-pypi.yml
docs/Makefile
docs/make.bat
docs/requirements.txt
docs/source/conf.py
docs/source/index.rst
docs/source/installation.rst
docs/source/reference.rst
docs/source/notebooks/intro.ipynb
docs/source/notebooks/smf_gradient_descent.png
docs/source/notebooks/smf_gradient_descent.py
multigrad/__init__.py
multigrad/_version.py
multigrad/adam.py
multigrad/bfgs.py
multigrad/multigrad.py
multigrad/util.py
multigrad.egg-info/PKG-INFO
multigrad.egg-info/SOURCES.txt
multigrad.egg-info/dependency_links.txt
multigrad.egg-info/requires.txt
multigrad.egg-info/top_level.txt
multigrad/diffdesi_experimental/util.py
multigrad/mpi4jax/__init__.py
multigrad/mpi4jax/multigrad.py
tests/__init__.py
tests/test_mpi.py
tests/smf_example/__init__.py
tests/smf_example/benchmark.py
tests/smf_example/gd_loss.png
tests/smf_example/gd_param.png
tests/smf_example/gd_param_path.png
tests/smf_example/hmf_model.png
tests/smf_example/parallel_sum_mpi4py_demo.py
tests/smf_example/smf_fit.png
tests/smf_example/smf_grad_descent.py
tests/smf_example/submit_benchmark_jobs.py