Loading repository data…
instadeepai / repository
jumanji
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
#jax#python#reinforcement-learning#research
Live repository statistics
★ 851Stars
⑂ 97Forks
◯ 31Open issues
◉ 851Watchers
README preview
Environments | Installation | Quickstart | Training | Citation | Docs
Jumanji @ ICLR 2024
Jumanji has been accepted at ICLR 2024, check out our research paper.
Welcome to the Jungle! 🌴
Jumanji is a diverse suite of scalable reinforcement learning environments written in JAX. It now features 22 environments!
Jumanji is helping pioneer a new wave of hardware-accelerated research and development in the field of RL. Jumanji's high-speed environments enable faster iteration and large-scale experimentation while simultaneously reducing complexity. Originating in the research team at InstaDeep, Jumanji is now developed jointly with the open-source community. To join us in these efforts, reach out, raise issues and read our contribution guidelines or just star 🌟 to stay up to date with the latest developments!
Goals 🚀
- Provide a simple, well-tested API for JAX-based environments.
- Make research in RL more accessible.
- Facilitate the research on RL for problems in the industry and help close the gap between research and industrial applications.
- Provide environments whose difficulty can be scaled to be arbitrarily hard.
Overview 🦜
- 🥑 Environment API: core abstractions for JAX-based environments.
- 🕹️ Environment Suite: a collection of RL environments ranging from simple games to NP-hard combinatorial problems.
- 🍬 Wrappers: easily connect to your favourite RL frameworks and libraries such as
Acme,
Stable Baselines3,
RLlib, Gymnasium
and DeepMind-Env through our
dm_envandgymwrappers. - 🎓 Examples: guides to facilitate Jumanji's adoption and highlight the added value of JAX-based environments.
- 🏎️ Training: example agents that can be used as inspiration for the agents one may implement in their research.
Jumanji provides a diverse range of environments ranging from simple games to NP-hard combinatorial problems.
| Environment | Category | Registered Version(s) | Source | Description |
|---|---|---|---|---|
| 🔢 Game2048 | Logic | Game2048-v1 | code | doc |
| 🎨 GraphColoring | Logic | GraphColoring-v1 | code | doc |
| 💣 Minesweeper | Logic | Minesweeper-v0 | code | doc |
| 🎲 RubiksCube | Logic | RubiksCube-v0RubiksCube-partly-scrambled-v0 | code | doc |
| 🔀 SlidingTilePuzzle | Logic | SlidingTilePuzzle-v0 | code | doc |
| ✏️ Sudoku | Logic | Sudoku-v0 Sudoku-very-easy-v0 | code | doc |
| 📦 BinPack (3D BinPacking Problem) | Packing | BinPack-v1 |