pytorch-labs / LeanRL
LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
â495Updated 3 months ago
Alternatives and similar repositories for LeanRL:
Users that are interested in LeanRL are comparing it to the libraries listed below
- Multi-Agent Reinforcement Learning with JAXâ487Updated this week
- Really Fast End-to-End Jax RL Implementationsâ800Updated 4 months ago
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ267Updated 3 weeks ago
- Distributed Reinforcement Learning accelerated by Lightning Fabricâ340Updated this week
- â178Updated last month
- Cost aware hyperparameter tuning algorithmâ140Updated 7 months ago
- RL Environments in JAX đâ682Updated 6 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ381Updated this week
- Accelerated minigrid environments with JAXâ128Updated 5 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â230Updated last week
- Efficient baselines for autocurricula in JAX.â176Updated 5 months ago
- Simple single-file baselines for Q-Learning in pure-GPU settingâ135Updated 2 months ago
- â211Updated 2 months ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.â651Updated 2 years ago
- Deep reinforcement learning without experience replay, target networks, or batch updates.â208Updated 2 weeks ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ220Updated last week
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilitiesâ335Updated 3 weeks ago
- Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmindâ359Updated 3 weeks ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ680Updated last month
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.â635Updated this week
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.â799Updated 5 months ago
- Reinforcement learning on general 2D physics environments in JAX.â121Updated this week
- GPU-acceleration of Nocturne via Madronaâ253Updated this week
- fast + parallel AlphaZero in JAXâ90Updated last month
- âī¸ Vectorized RL game environments in JAXâ435Updated last month
- Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Labâ630Updated this week
- MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Researchâ487Updated 5 months ago
- Kolmogorov-Arnold Network for Reinforcement Leaning, initial experimentsâ272Updated 6 months ago
- Modular framework for Reinforcement Learning in pythonâ170Updated last year
- A reinforcement learning framework based on MLX.â224Updated 11 months ago