Emerge-Lab / gpudrive
GPU-acceleration of Nocturne via Madrona
☆232Updated this week
Related projects ⓘ
Alternatives and complementary repositories for gpudrive
- A data-driven, fast driving simulator for multi-agent coordination under partial observability.☆264Updated 5 months ago
- ☆66Updated 11 months ago
- Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind☆348Updated last week
- Multi-Agent Reinforcement Learning with JAX☆441Updated 2 weeks ago
- Reinforcement learning on general 2D physics environments in JAX☆106Updated last week
- Efficient baselines for autocurricula in JAX.☆173Updated 2 months ago
- Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"☆394Updated last week
- ☆320Updated this week
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆454Updated 3 weeks ago
- ☆234Updated last year
- Official implementation of the RLC 2024 paper "Policy-Guided Diffusion"☆119Updated 4 months ago
- A 3D video game environment and benchmark designed from scratch for reinforcement learning research☆181Updated last year
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆149Updated last year
- Accelerated minigrid environments with JAX☆119Updated 3 months ago
- An engine for high performance multi-agent environments with very large numbers of agents, along with a set of reference environments☆230Updated 2 weeks ago
- ☆177Updated 3 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆84Updated 2 months ago
- Really Fast End-to-End Jax RL Implementations☆735Updated 2 months ago
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆323Updated 5 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆208Updated last month
- DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor Control☆88Updated 3 weeks ago
- ☆63Updated 3 months ago
- VMAS is a vectorized differentiable simulator designed for efficient Multi-Agent Reinforcement Learning benchmarking. It is comprised of …☆346Updated 2 weeks ago
- ☆149Updated this week
- Benchmarks for Multi-Objective Multi-Agent Decision Making☆72Updated 3 weeks ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆209Updated last year
- Implementation of Dreamer v3 in pytorch.☆428Updated last month
- ☆245Updated 2 years ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆239Updated last week
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆213Updated 3 weeks ago