Steps: [ install | jax | haiku | q-learning | dqn | ppo | next_steps ]
Install¶
Coax is built on top of JAX, but it doesn’t have an explicit dependence on the jax
python package. The reason is that your version of jaxlib
will depend on your CUDA version.
To install jax
, please have a look at the instructions: https://github.com/google/jax#installation
Once jax
and jaxlib
are installed, you can install coax simple by running:
$ pip install coax
Or, alternatively, to install coax from the latest branch on github:
$ pip install git+https://github.com/coax-dev/coax.git@main