Emerge-Lab / gpudrive
GPU-acceleration of Nocturne via Madrona
☆250Updated 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.☆270Updated 7 months ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆489Updated 2 months ago
- ☆69Updated last year
- Deep reinforcement learning without experience replay, target networks, or batch updates.☆206Updated last week
- Efficient baselines for autocurricula in JAX.☆175Updated 4 months ago
- Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind☆359Updated 2 weeks ago
- ☆181Updated 5 months ago
- Reinforcement learning on general 2D physics environments in JAX.☆121Updated last month
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆264Updated 2 weeks ago
- [ECCV'24] SLEDGE: Synthesizing Driving Environments with Generative Models and Rule-Based Traffic☆167Updated 3 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆89Updated 3 months ago
- ☆177Updated last month
- Official implementation of the RLC 2024 paper "Policy-Guided Diffusion"☆127Updated 6 months ago
- Multi-Agent Reinforcement Learning with JAX☆482Updated last week
- Simple single-file baselines for Q-Learning in pure-GPU setting☆133Updated last month
- Really Fast End-to-End Jax RL Implementations☆797Updated 4 months ago
- ☆337Updated this week
- Accelerated minigrid environments with JAX☆127Updated 5 months ago
- A 3D video game environment and benchmark designed from scratch for reinforcement learning research☆181Updated last year
- Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"☆429Updated 3 weeks ago
- Implementation of π₀, the robotic foundation model architecture proposed by Physical Intelligence☆298Updated this week
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆349Updated 7 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆228Updated 3 months ago
- Implementation of Dreamer v3 in pytorch.☆465Updated 3 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆219Updated last week
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆148Updated last year
- VMAS is a vectorized differentiable simulator designed for efficient Multi-Agent Reinforcement Learning benchmarking. It is comprised of …☆366Updated 3 weeks ago
- Benchmarks for Multi-Objective Multi-Agent Decision Making☆77Updated 2 months ago
- ☆254Updated last year
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.☆128Updated 5 months ago