jaxpm-paper

Repo for paper on Lagragian Neural ODEs for fast N-body simulations

To run notebooks, first install the following dependencies:

pip install git+https://github.com/DifferentiableUniverseInitiative/JaxPM.git dm-haiku diffrax

You also need to find a way to install python-fastpm... not always easy