jumanji
RL Environments
A suite of scalable reinforcement learning environments
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
657 stars
12 watching
83 forks
Language: Python
last commit: 4 months ago
Linked from 3 awesome lists
jaxpythonreinforcement-learningresearch
ethicalml/awesome-production-machine-learning
aikorea/awesome-rl
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
| instadeepai/mava | A research-friendly codebase for experimenting with multi-agent reinforcement learning in JAX | 749 |
 juliareinforcementlearning/reinforcementlearninganintroduction.jl | Provides Julia code to generate figures and examples for an introductory reinforcement learning book | 309 |
 google-deepmind/spriteworld | An environment designed to test and train reinforcement learning algorithms in a flexible, procedurally generated 2D space with various objects and interactions. | 369 |
| google-deepmind/android_env | A platform for defining reinforcement learning tasks on top of Android devices | 1,029 |
 luchris429/purejaxrl | A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality | 755 |
 jjkke88/rl_toolbox | A collection of reinforcement learning algorithms and tools for training agents in complex environments. | 43 |
 ikostrikov/jaxrl | Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces. | 640 |
 matyama/deep-rl-hands-on | A collection of examples and exercises from a deep reinforcement learning book | 2 |
| google-deepmind/meltingpot | Assesses generalization of multi-agent reinforcement learning algorithms to novel social situations | 637 |
 toni-sm/skrl | A modular reinforcement learning library with support for various environments and frameworks | 588 |
 intelligent-environments-lab/citylearn | A reinforcement learning environment for simulating energy management in buildings and neighborhoods. | 480 |
 google-research/deep_ope | Provides benchmarking policies and datasets for offline reinforcement learning | 85 |
 geek-ai/magent | A platform for multi-agent reinforcement learning research and development | 1,700 |
 matthiasplappert/keras-rl | A Python library implementing state-of-the-art deep reinforcement learning algorithms for Keras and OpenAI Gym environments. | 8 |
 farama-foundation/d4rl | A collection of standardized environments and datasets for training and benchmarking algorithms in offline reinforcement learning | 1,371 |