gymnax
by RobertTLange
RL Environments in JAX 🌍
AI summary
RL accelerator
A framework that accelerates RL environment processes by leveraging JAX and GPU acceleration
- stars
- 669
- forks
- 62
- watching
- 10
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
google/objax772
Research accelerator
An open-source machine learning framework designed to simplify and accelerate research
RL framework
A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality
RL builder
A Python library providing reinforcement learning building blocks for implementing agent policies and functions
kaixhin/rainbow1.6K
RL framework
A Python implementation of a deep reinforcement learning algorithm combining multiple techniques for improved performance in Atari games
RL framework
A PyTorch framework for accelerating reinforcement learning research and development by providing a modular, reusable, and customizable training loop
RL framework
A modular framework for building reinforcement learning agents in Python using Gymnasium and JAX.
toni-sm/skrl588
RL library
A modular reinforcement learning library with support for various environments and frameworks
RL toolkit
Provides a unified toolkit for constructing, computing, and optimizing intrinsic reward modules in reinforcement learning
RL framework
A Python library implementing state-of-the-art deep reinforcement learning algorithms for Keras and OpenAI Gym environments.
LLM trainer
A flexible RL training framework designed for large language models
RL toolkit
A toolkit for developing and evaluating reinforcement learning algorithms in a reproducible manner
RL framework
An open-source reinforcement learning framework for autonomous driving tasks using the Carla-Simulator environment and Ray/Rllib libraries.
RL Framework
A framework for distributed reinforcement learning research with Python and PyTorch support
geek-ai/magent1.7K
RL platform
A platform for multi-agent reinforcement learning research and development
RL frameworks
A collection of implementations of Reinforcement Learning and planning algorithms in Python.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions gymnax.