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