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