gymnax
RL accelerator
A framework that accelerates RL environment processes by leveraging JAX and GPU acceleration
RL Environments in JAX 🌍
669 stars
10 watching
62 forks
Language: Python
last commit: 6 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 | 772 |
 luchris429/purejaxrl | A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality | 755 |
 google-deepmind/rlax | A Python library providing reinforcement learning building blocks for implementing agent policies and functions | 1,272 |
 kaixhin/rainbow | A Python implementation of a deep reinforcement learning algorithm combining multiple techniques for improved performance in Atari games | 1,591 |
 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. | 168 |
 toni-sm/skrl | A modular reinforcement learning library with support for various environments and frameworks | 588 |
 rle-foundation/rlexplore | Provides a unified toolkit for constructing, computing, and optimizing intrinsic reward modules in reinforcement learning | 373 |
 matthiasplappert/keras-rl | A Python library implementing state-of-the-art deep reinforcement learning algorithms for Keras and OpenAI Gym environments. | 8 |
 volcengine/verl | A flexible RL training framework designed for large language models | 427 |
 rlworkgroup/garage | A toolkit for developing and evaluating reinforcement learning algorithms in a reproducible manner | 1,893 |
 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,700 |
 eleurent/rl-agents | A collection of implementations of Reinforcement Learning and planning algorithms in Python. | 596 |