Emerge-Lab / gpudrive
1 million FPS multi-agent driving simulator
☆368Updated this week
Alternatives and similar repositories for gpudrive:
Users that are interested in gpudrive are comparing it to the libraries listed below
- A data-driven, fast driving simulator for multi-agent coordination under partial observability.☆273Updated 9 months ago
- Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"☆475Updated 3 weeks ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆524Updated 4 months ago
- Multi-Agent Reinforcement Learning with JAX☆541Updated 2 weeks ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆170Updated this week
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆294Updated last month
- Deep reinforcement learning without experience replay, target networks, or batch updates.☆225Updated this week
- Implementation of Dreamer v3 in pytorch.☆511Updated 5 months ago
- ☆186Updated 7 months ago
- ☆369Updated this week
- PyTorch code for the paper "Model-Based Imitation Learning for Urban Driving".☆384Updated last year
- [ECCV'24] SLEDGE: Synthesizing Driving Environments with Generative Models and Rule-Based Traffic☆174Updated 5 months ago
- ☆191Updated 3 months ago
- VMAS is a vectorized differentiable simulator designed for efficient Multi-Agent Reinforcement Learning benchmarking. It is comprised of …☆400Updated 3 weeks ago
- Distributed Reinforcement Learning accelerated by Lightning Fabric☆355Updated this week
- Really Fast End-to-End Jax RL Implementations☆841Updated 6 months ago
- Accelerated minigrid environments with JAX☆132Updated 7 months ago
- ☆70Updated last year
- Simple single-file baselines for Q-Learning in pure-GPU setting☆149Updated this week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆423Updated 2 weeks ago
- DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor Control☆99Updated 4 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆300Updated this week
- RL Environments in JAX 🌍☆722Updated 8 months ago
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆150Updated last year
- Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind☆368Updated last month
- ☆268Updated 2 years ago
- Mastering Diverse Domains through World Models☆1,552Updated last month
- A collection of MARL benchmarks based on TorchRL☆366Updated this week