jumanji
RL environments
Provides a suite of scalable reinforcement learning environments for research and development.
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
622 stars
12 watching
80 forks
Language: Python
last commit: 6 days 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 | 734 |
 juliareinforcementlearning/reinforcementlearninganintroduction.jl | Provides Julia code to generate figures and examples for an introductory reinforcement learning book | 310 |
 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. | 368 |
| google-deepmind/android_env | A platform for defining reinforcement learning tasks on top of Android devices | 1,019 |
 luchris429/purejaxrl | A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality | 722 |
 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. | 630 |
 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 | 620 |
 toni-sm/skrl | A modular reinforcement learning library with support for various environments and frameworks | 560 |
 intelligent-environments-lab/citylearn | A reinforcement learning environment for simulating energy management in buildings and neighborhoods. | 478 |
 google-research/deep_ope | A set of pre-trained reinforcement learning policies and benchmarking data for offline model selection in reinforcement learning. | 85 |
 geek-ai/magent | A platform for multi-agent reinforcement learning research and development | 1,690 |
 matthiasplappert/keras-rl | A Python library implementing state-of-the-art deep reinforcement learning algorithms for Keras and OpenAI Gym environments. | 7 |
 farama-foundation/d4rl | Provides standardized environments and datasets for offline reinforcement learning training and benchmarking | 1,346 |