Installation

Make sure you have JAX and the usual suspects with,

pip install jax jaxlib numpy matplotlib scipy

optionally include packages scikit-learn and dm-haiku if running some examples.

Install the stable version with,

pip install stick_ray

or the latest release with

pip install git+http://github.com/Joshuaalbert/stick_ray.git