andyljones / megastepView external linksLinks
megastep helps you build 1-million FPS reinforcement learning environments on a single GPU
☆146Feb 9, 2022Updated 4 years ago
Alternatives and similar repositories for megastep
Users that are interested in megastep are comparing it to the libraries listed below
Sorting:
- Standard interface for entity based reinforcement learning environments.☆38Feb 28, 2024Updated last year
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- RL agent to play μRTS with Stable-Baselines3 and PyTorch☆27Jan 23, 2022Updated 4 years ago
- Repo for the multi-agent PressurePlate environment☆17Feb 4, 2022Updated 4 years ago
- OpenAi's gym environment wrapper to vectorize them with Ray☆23May 25, 2023Updated 2 years ago
- A grid-world game engine for game AI research☆253Apr 9, 2024Updated last year
- ☆327Dec 19, 2024Updated last year
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,269Aug 12, 2024Updated last year
- ☆10Jun 27, 2024Updated last year
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- RL Environments in JAX 🌍☆857May 30, 2025Updated 8 months ago
- Codebase for BRDiv: Diverse teammate generation for ad hoc teamwork☆13May 2, 2024Updated last year
- ☆59Sep 22, 2022Updated 3 years ago
- ☆21Dec 22, 2020Updated 5 years ago
- Deep reinforcement learning implementation that trains AIs for the CodeCraft real-time strategy game.☆21May 22, 2023Updated 2 years ago
- Opinionated library for managing hyperparameters and mutable state of machine learning training systems.☆19Aug 4, 2023Updated 2 years ago
- Procgen Benchmark: Procedurally-Generated Game-Like Gym-Environments☆1,136Jan 3, 2024Updated 2 years ago
- Baselines for gymnax 🤖☆74Apr 3, 2023Updated 2 years ago
- DeepMind Alchemy task environment: a meta-reinforcement learning benchmark☆205Apr 5, 2023Updated 2 years ago
- High throughput synchronous and asynchronous reinforcement learning☆973Jan 29, 2026Updated 2 weeks ago
- Evaluating long-term memory of reinforcement learning algorithms☆163Jun 23, 2023Updated 2 years ago
- ☆13Aug 9, 2022Updated 3 years ago
- A toolkit for practical Human-AI cooperation research☆14Apr 19, 2024Updated last year
- A 3D video game environment and benchmark designed from scratch for reinforcement learning research☆190May 3, 2023Updated 2 years ago
- A C++ pytorch implementation of MuZero☆40May 1, 2024Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆59Aug 4, 2022Updated 3 years ago
- Train an agent to play VizDoom with multi sensory inputs. Trained using sample factory☆14Jul 9, 2021Updated 4 years ago
- Collection of in-progress libraries for entity neural networks.☆30Jun 24, 2022Updated 3 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆122Aug 22, 2024Updated last year
- General Modules for JAX☆72Sep 12, 2025Updated 5 months ago
- ☆91Jan 21, 2026Updated 3 weeks ago
- Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)☆499May 1, 2025Updated 9 months ago
- A standalone release of DeepMind Lab's maze generator with Python bindings.☆67Oct 3, 2023Updated 2 years ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆371Jul 7, 2025Updated 7 months ago
- Massively parallel rigidbody physics simulation on accelerator hardware.☆3,058Updated this week
- ☆18Apr 22, 2024Updated last year
- The Starcraft Multi-Agent challenge lite☆47Sep 13, 2024Updated last year
- A simple OpenAI Gym environment for single and multi-agent reinforcement learning☆774Dec 14, 2023Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago