A Python interface for reinforcement learning environments
☆394Dec 23, 2022Updated 3 years ago
Alternatives and similar repositories for dm_env
Users that are interested in dm_env are comparing it to the libraries listed below
Sorting:
- ☆1,411Mar 2, 2026Updated 2 weeks ago
- A library of reinforcement learning components and agents☆3,946Mar 12, 2026Updated last week
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,531Apr 13, 2024Updated last year
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research☆768Mar 9, 2026Updated last week
- A tool for recording RL trajectories.☆112Mar 11, 2026Updated last week
- A suite of test scenarios for multi-agent reinforcement learning.☆801Mar 12, 2026Updated last week
- RL Environments in JAX 🌍☆871May 30, 2025Updated 9 months ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆61Oct 23, 2023Updated 2 years ago
- A collection of RL algorithms written in JAX.☆105Jul 5, 2022Updated 3 years ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆753Oct 26, 2022Updated 3 years ago
- JAX-based neural network library☆3,197Mar 12, 2026Updated last week
- Massively parallel rigidbody physics simulation on accelerator hardware.☆3,094Updated this week
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆495Apr 6, 2024Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆19Jul 9, 2024Updated last year
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.☆4,498Updated this week
- ☆929Mar 12, 2026Updated last week
- A networking protocol for agent-environment communication☆108Feb 20, 2026Updated last month
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆867Aug 12, 2024Updated last year
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆274Sep 22, 2025Updated 5 months ago
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- 🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX☆885Feb 26, 2026Updated 3 weeks ago
- A customisable 2D platform for agent-based AI research☆439Oct 5, 2023Updated 2 years ago
- General Modules for JAX☆72Feb 21, 2026Updated 3 weeks ago
- OpenAI Gym wrapper for the DeepMind Control Suite☆229May 19, 2024Updated last year
- Procgen Benchmark: Procedurally-Generated Game-Like Gym-Environments☆1,147Jan 3, 2024Updated 2 years ago
- Optax is a gradient processing and optimization library for JAX.☆2,212Updated this week
- Libraries, tools and tasks created and used at DeepMind Robotics.☆408Aug 6, 2025Updated 7 months ago
- A collection of wrappers for Gymnasium and PettingZoo environments (being merged into gymnasium.wrappers and pettingzoo.wrappers☆491Aug 11, 2025Updated 7 months ago
- Library for Model Based RL☆1,058Jul 12, 2024Updated last year
- Really Fast End-to-End Jax RL Implementations☆1,032Sep 9, 2024Updated last year
- Baselines for gymnax 🤖☆74Apr 3, 2023Updated 2 years ago
- ☆47Sep 24, 2024Updated last year
- Fast reinforcement learning research☆61Dec 7, 2024Updated last year
- A standalone release of DeepMind Lab's maze generator with Python bindings.☆67Oct 3, 2023Updated 2 years ago
- ☆164Dec 13, 2023Updated 2 years ago
- A PyTorch Platform for Distributed RL☆751Sep 15, 2021Updated 4 years ago
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,279Aug 12, 2024Updated last year
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆15Jan 3, 2023Updated 3 years ago