purejaxrl
RL framework
A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality
Really Fast End-to-End Jax RL Implementations
755 stars
12 watching
63 forks
Language: Python
last commit: 6 months ago
Linked from 1 awesome list
deep-reinforcement-learningjaxpporeinforcement-learningreinforcement-learning-algorithms
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 matthiasplappert/keras-rl | A Python library implementing state-of-the-art deep reinforcement learning algorithms for Keras and OpenAI Gym environments. | 8 |
 kaixhin/rainbow | A Python implementation of a deep reinforcement learning algorithm combining multiple techniques for improved performance in Atari games | 1,591 |
 iffix/machin | An open-source reinforcement learning library for PyTorch, providing a simple and clear implementation of various algorithms. | 402 |
 ikostrikov/jaxrl | Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces. | 640 |
 astooke/rlpyt | A modular and unified framework for implementing common deep reinforcement learning algorithms in PyTorch | 2,236 |
 jingweiz/pytorch-rl | A comprehensive project that provides an implementation of deep reinforcement learning algorithms using PyTorch and Visdom. | 798 |
 catalyst-team/catalyst-rl | A PyTorch framework for accelerating reinforcement learning research and development by providing a modular, reusable, and customizable training loop | 46 |
 kunqian2025/reinforcement-learning | A collection of implementations of reinforcement learning algorithms in MATLAB | 61 |
 scitator/catalyst-rl-framework | A framework for distributed reinforcement learning research with Python and PyTorch support | 38 |
 sjtu-marl/malib | A framework for parallel population-based reinforcement learning | 507 |
 enlite-ai/maze | An RL framework for building and training reinforcement learning models in Python | 266 |
 coax-dev/coax | A modular framework for building reinforcement learning agents in Python using Gymnasium and JAX. | 168 |
 layssi/carla_ray_rlib | An open-source reinforcement learning framework for autonomous driving tasks using the Carla-Simulator environment and Ray/Rllib libraries. | 35 |
 rl-tools/rl-tools | An open-source C++ library for training reinforcement learning policies and controlling continuous-control environments. | 680 |
 jjkke88/rl_toolbox | A collection of reinforcement learning algorithms and tools for training agents in complex environments. | 43 |