coax
RL framework
A modular framework for building reinforcement learning agents in Python using Gymnasium and JAX.
Modular framework for Reinforcement Learning in python
167 stars
8 watching
17 forks
Language: Python
last commit: almost 2 years ago
Linked from 1 awesome list
reinforcement-learningreinforcement-learning-agentreinforcement-learning-algorithms
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 luchris429/purejaxrl | A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality | 722 |
 eleurent/rl-agents | A collection of implementations of Reinforcement Learning and planning algorithms in Python. | 589 |
 kunqian2025/reinforcement-learning | A collection of implementations of reinforcement learning algorithms in MATLAB | 60 |
 sjtu-marl/malib | A framework for parallel population-based reinforcement learning | 497 |
 kaixhin/rainbow | A Python implementation of a deep reinforcement learning algorithm combining multiple techniques for improved performance in Atari games | 1,585 |
 enlite-ai/maze | An RL framework for building and training reinforcement learning models in Python | 265 |
 scitator/catalyst-rl-framework | A framework for distributed reinforcement learning research with Python and PyTorch support | 38 |
 catalyst-team/catalyst-rl | A PyTorch framework for accelerating reinforcement learning research and development by providing a modular, reusable, and customizable training loop | 46 |
 iffix/machin | An open-source reinforcement learning library for PyTorch, providing a simple and clear implementation of various algorithms. | 401 |
 layssi/carla_ray_rlib | An open-source reinforcement learning framework for autonomous driving tasks using the Carla-Simulator environment and Ray/Rllib libraries. | 35 |
 horizonrobotics/alf | A reinforcement learning framework designed to implement complex algorithms with flexibility and ease of use | 302 |
 gokulnc/setting-up-carla-reinforcement-learning | Provides a framework for using CARLA as a reinforcement learning environment | 95 |
 jjkke88/rl_toolbox | A collection of reinforcement learning algorithms and tools for training agents in complex environments. | 43 |
 matthiasplappert/keras-rl | A Python library implementing state-of-the-art deep reinforcement learning algorithms for Keras and OpenAI Gym environments. | 7 |
 tatsu-lab/alpaca_farm | A framework for simulating and evaluating reinforcement learning from human feedback methods | 782 |