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