shacklettbp / madrona
☆369Updated this week
Alternatives and similar repositories for madrona:
Users that are interested in madrona are comparing it to the libraries listed below
- [ICLR 2022] Accelerated Policy Learning with Parallel Differentiable Simulation☆303Updated 10 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆295Updated last month
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆525Updated 5 months ago
- ☆21Updated 5 months ago
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,134Updated 7 months ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆170Updated this week
- Really Fast End-to-End Jax RL Implementations☆840Updated 6 months ago
- RL Environments in JAX 🌍☆722Updated 8 months ago
- Multi-Agent Reinforcement Learning with JAX☆545Updated 3 weeks ago
- High-throughput simulation platform for Artificial Intelligence reseach☆222Updated 2 years ago
- ☆191Updated 3 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆424Updated 2 weeks ago
- Efficient baselines for autocurricula in JAX.☆186Updated 7 months ago
- megastep helps you build 1-million FPS reinforcement learning environments on a single GPU☆133Updated 3 years ago
- ☆256Updated 2 years ago
- A minimal and stable PPO.☆134Updated last year
- ☆218Updated 4 months ago
- Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab☆698Updated this week
- Official implementation of Implicit Behavioral Cloning, as described in our CoRL 2021 paper, see more at https://implicitbc.github.io/☆338Updated last year
- An API conversion tool for popular external reinforcement learning environments☆157Updated 2 months ago
- Simulate and Render MuJoCo Models in the Browser!☆228Updated 9 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆300Updated this week
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆228Updated last week
- ☆70Updated last year
- Simple single-file baselines for Q-Learning in pure-GPU setting☆150Updated last week
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆669Updated 2 years ago
- An open-source library for GPU-accelerated robot learning and sim-to-real transfer.☆855Updated this week
- Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind☆368Updated last month
- DayDreamer: World Models for Physical Robot Learning☆305Updated 2 years ago
- ♟️ Vectorized RL game environments in JAX☆456Updated 3 weeks ago