Inference on non-linear dynamical systems written in JAX.

Requires python 3.X.

As always I reccomend a package manager like conda. In this case you could do conda create -n jaxnlds_py python=3.8 && conda activate jaxnlds_py

Install usual suspects: pip install numpy scipy matplotlib jax jaxlib

If you want nested sampling: pip install git+https://github.com/Joshuaalbert/jaxns.git

Then install this with python setup.py install or pip install git+https://github.com/Joshuaalbert/jaxnlds.git