Need help with coax?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

158 Stars 14 Forks MIT License 302 Commits 0 Opened issues


Turn RL papers into code, the easy way

Services available


Need anything else?

Contributors list

|tests| |pypi| |docs| |License|


Plug-n-Play Reinforcement Learning in Python with

OpenAI Gym 
_ and


For the full documentation, including many examples, go to


coax is built on top of JAX, but it doesn't have an explicit dependence on the

python package. The reason is that your version of
will depend on your CUDA version. To install without CUDA, simply run:

.. code-block::

$ pip install jaxlib jax coax --upgrade

If you do require CUDA support, please check out the

Installation Guide 

Getting Started

Have a look at the

Getting Started 
_ page to train your first RL agent.


This project welcomes contributions and suggestions. Most contributions require you to agree to a Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us the rights to use your contribution. For details, visit

When you submit a pull request, a CLA bot will automatically determine whether you need to provide a CLA and decorate the PR appropriately (e.g., status check, comment). Simply follow the instructions provided by the bot. You will only need to do this once across all repos using our CLA.

This project has adopted the

Microsoft Open Source Code of Conduct 
. For more information see the
Code of Conduct FAQ 
or contact
[email protected] 
_ with any additional questions or comments.


.. |readthedocs| image:: :target: :width: 400 :alt: readthedocs

.. |tests| image:: :target: :alt: tests badge

.. |pypi| image:: :target: :alt: pypi badge

.. |docs| image:: :target: :alt: docs badge

.. |license| image:: :target: :alt: license badge

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.