https://github.com/coax-dev/coax

Modular framework for Reinforcement Learning in python
https://github.com/coax-dev/coax

reinforcement-learning reinforcement-learning-agent reinforcement-learning-algorithms

Last synced: 3 months ago
JSON representation

Modular framework for Reinforcement Learning in python

• Host: GitHub
• URL: https://github.com/coax-dev/coax
• Owner: coax-dev
• License: mit
• Created: 2021-03-11T00:56:08.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2023-02-01T18:54:38.000Z (almost 2 years ago)
• Last Synced: 2024-07-12T02:38:52.701Z (4 months ago)
• Topics: reinforcement-learning, reinforcement-learning-agent, reinforcement-learning-algorithms
• Language: Python
• Homepage: https://coax.readthedocs.io
• Size: 17.6 MB
• Stars: 164
• Watchers: 8
• Forks: 18
• Open Issues: 7
• Metadata Files:
  • Readme: README.rst
  • License: LICENSE
  • Codeowners: .github/CODEOWNERS

Awesome Lists containing this project

• awesome-jax - Coax - Turn RL papers into code, the easy way. <img src="https://img.shields.io/github/stars/coax-dev/coax?style=social" align="center"> (Libraries)

README

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

coax

====

Plug-n-Play Reinforcement Learning in Python with `Gymnasium `_

(formerly OpenAI Gym) and `JAX `_

|readthedocs|

For the full documentation, including many examples, go to https://coax.readthedocs.io/

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

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.

....................................................................................................

.. |readthedocs| image:: https://raw.githubusercontent.com/coax-dev/coax/main/doc/_static/img/readthedocs.gif

    :target: https://coax.readthedocs.io/

    :width: 400

    :alt: readthedocs

.. |tests| image:: https://github.com/coax-dev/coax/workflows/tests/badge.svg

    :target: https://github.com/coax-dev/coax/actions?query=workflow%3Atests

    :alt: tests badge

.. |pypi| image:: https://img.shields.io/pypi/v/coax

    :target: https://pypi.org/project/coax

    :alt: pypi badge

.. |docs| image:: https://readthedocs.org/projects/coax/badge/?version=latest

    :target: https://coax.readthedocs.io

    :alt: docs badge

.. |license| image:: https://img.shields.io/github/license/coax-dev/coax

    :target: https://github.com/coax-dev/coax/blob/main/LICENSE

    :alt: license badge