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