gymnax
RL accelerator
A framework that accelerates RL environment processes by leveraging JAX and GPU acceleration
RL Environments in JAX 🌍
650 stars
10 watching
62 forks
Language: Python
last commit: 5 months ago
Linked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 google/objax | An open-source machine learning framework designed to simplify and accelerate research | 771 |
 luchris429/purejaxrl | A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality | 722 |
 google-deepmind/rlax | A Python library providing reinforcement learning building blocks for implementing agent policies and functions | 1,263 |
 kaixhin/rainbow | A Python implementation of a deep reinforcement learning algorithm combining multiple techniques for improved performance in Atari games | 1,585 |
 catalyst-team/catalyst-rl | A PyTorch framework for accelerating reinforcement learning research and development by providing a modular, reusable, and customizable training loop | 46 |
 coax-dev/coax | A modular framework for building reinforcement learning agents in Python using Gymnasium and JAX. | 167 |
 toni-sm/skrl | A modular reinforcement learning library with support for various environments and frameworks | 560 |
 rle-foundation/rlexplore | Provides a unified toolkit for constructing, computing, and optimizing intrinsic reward modules in reinforcement learning | 366 |
 matthiasplappert/keras-rl | A Python library implementing state-of-the-art deep reinforcement learning algorithms for Keras and OpenAI Gym environments. | 7 |
 volcengine/verl | A flexible and efficient reinforcement learning framework designed for large language models. | 315 |
 rlworkgroup/garage | A toolkit for developing and evaluating reinforcement learning algorithms in a reproducible manner | 1,880 |
 layssi/carla_ray_rlib | An open-source reinforcement learning framework for autonomous driving tasks using the Carla-Simulator environment and Ray/Rllib libraries. | 35 |
 scitator/catalyst-rl-framework | A framework for distributed reinforcement learning research with Python and PyTorch support | 38 |
 geek-ai/magent | A platform for multi-agent reinforcement learning research and development | 1,690 |
 eleurent/rl-agents | A collection of implementations of Reinforcement Learning and planning algorithms in Python. | 589 |