Emerge-Lab / gpudrive
1 million FPS multi-agent driving simulator
☆272Updated 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.☆271Updated 8 months ago
- [ECCV'24] SLEDGE: Synthesizing Driving Environments with Generative Models and Rule-Based Traffic☆171Updated 4 months ago
- ☆182Updated 6 months ago
- Deep reinforcement learning without experience replay, target networks, or batch updates.☆213Updated 3 weeks ago
- ☆70Updated last year
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆129Updated last week
- Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"☆448Updated 2 weeks ago
- Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind☆365Updated last week
- ☆348Updated this week
- Accelerated minigrid environments with JAX☆128Updated 6 months ago
- PyTorch code for the paper "Model-Based Imitation Learning for Urban Driving".☆375Updated last year
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆148Updated last year
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆507Updated 3 months ago
- ☆188Updated 2 months ago
- Implementation of Dreamer v3 in pytorch.☆489Updated 4 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆137Updated 2 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆280Updated last week
- Multi-Agent Reinforcement Learning with JAX☆509Updated last week
- Distributed Reinforcement Learning accelerated by Lightning Fabric☆345Updated last week
- A unified interface to many trajectory forecasting datasets.☆362Updated 2 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆277Updated this week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆396Updated this week
- PyTorch implementation of DreamerV2 model-based RL algorithm☆215Updated last year
- ☆259Updated 2 years ago
- Really Fast End-to-End Jax RL Implementations☆810Updated 5 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆92Updated 4 months ago
- RL Environments in JAX 🌍☆703Updated 7 months ago
- Efficient baselines for autocurricula in JAX.☆179Updated 5 months ago
- Official implementation of the RLC 2024 paper "Policy-Guided Diffusion"☆128Updated 7 months ago
- An OpenAI Gym environment for multi-agent car racing based on Gym's original car racing environment.☆79Updated 3 years ago