pytorch-labs / LeanRL
LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
☆432Updated 2 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for LeanRL
- Really Fast End-to-End Jax RL Implementations☆715Updated 2 months ago
- Distributed Reinforcement Learning accelerated by Lightning Fabric☆314Updated last week
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆230Updated 3 weeks ago
- Multi-Agent Reinforcement Learning with JAX☆438Updated this week
- RL Environments in JAX 🌍☆644Updated 4 months ago
- Cost aware hyperparameter tuning algorithm☆122Updated 4 months ago
- Efficient baselines for autocurricula in JAX.☆173Updated 2 months ago
- Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind☆341Updated last month
- fast + parallel AlphaZero in JAX☆84Updated 7 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆202Updated 3 weeks ago
- SBX: Stable Baselines Jax (SB3 + Jax)☆336Updated last week
- ☆147Updated 2 months ago
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.☆589Updated this week
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆617Updated last week
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️☆197Updated last month
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆625Updated 2 years ago
- Accelerated minigrid environments with JAX☆114Updated 3 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆210Updated 2 weeks ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆291Updated this week
- GPU-acceleration of Nocturne via Madrona☆227Updated this week
- Simple single-file baselines for Q-Learning in pure-GPU setting☆94Updated 3 months ago
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆314Updated 5 months ago
- ☆200Updated 9 months ago
- Kolmogorov-Arnold Network for Reinforcement Leaning, initial experiments☆269Updated 4 months ago
- ☆197Updated 3 months ago
- A suite of test scenarios for multi-agent reinforcement learning.☆616Updated last week
- Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab☆551Updated last week
- General multi-task deep RL Agent☆164Updated 5 months ago
- ☆223Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆252Updated 5 months ago