jaxrl
RL library
Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces.
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
630 stars
12 watching
69 forks
Language: Jupyter Notebook
last commit: about 2 years ago
Linked from 1 awesome list
batch-reinforcement-learningbehavioral-cloningcontinuous-controldeep-deterministic-policy-gradientdeep-learningdeep-reinforcement-learningflaxgymjaxoffline-reinforcement-learningreinforcement-learningsacsoft-actor-critic
Related projects:
| Repository | Description | Stars |
|---|---|---|
 luchris429/purejaxrl | A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality | 722 |
| ikostrikov/pytorch-ddpg-naf | An implementation of reinforcement learning algorithms for continuous control tasks using deep neural networks. | 307 |
 toni-sm/skrl | A modular reinforcement learning library with support for various environments and frameworks | 560 |
 matthiasplappert/keras-rl | A Python library implementing state-of-the-art deep reinforcement learning algorithms for Keras and OpenAI Gym environments. | 7 |
 rl-tools/rl-tools | A comprehensive deep reinforcement learning library for continuous control tasks | 216 |
 jjkke88/rl_toolbox | A collection of reinforcement learning algorithms and tools for training agents in complex environments. | 43 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 kaixhin/rainbow | A Python implementation of a deep reinforcement learning algorithm combining multiple techniques for improved performance in Atari games | 1,585 |
 n2cholas/jax-resnet | Provides implementations and checkpoints for various ResNet variants using JAX and Flax. | 104 |
 iffix/machin | An open-source reinforcement learning library for PyTorch, providing a simple and clear implementation of various algorithms. | 401 |
 cpnota/autonomous-learning-library | A comprehensive library for building deep reinforcement learning agents using PyTorch | 646 |
 corl-team/corl | Provides high-quality implementations of offline and offline-to-online reinforcement learning algorithms in Python. | 482 |
 kunqian2025/reinforcement-learning | A collection of implementations of reinforcement learning algorithms in MATLAB | 60 |
 catohaste/pomdp | An implementation of a reinforcement learning algorithm based on a partially observable Markov decision process for a simple decision task | 52 |
 takuseno/d3rlpy | An open-source deep reinforcement learning library that supports offline and online training, providing an intuitive API for practitioners and researchers. | 1,327 |