pytorch-labs / LeanRL
LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
â504Updated 3 months ago
Alternatives and similar repositories for LeanRL:
Users that are interested in LeanRL are comparing it to the libraries listed below
- Really Fast End-to-End Jax RL Implementationsâ808Updated 5 months ago
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ275Updated this week
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â275Updated this week
- â180Updated 2 months ago
- Distributed Reinforcement Learning accelerated by Lightning Fabricâ344Updated this week
- RL Environments in JAX đâ696Updated 7 months ago
- Multi-Agent Reinforcement Learning with JAXâ499Updated this week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ392Updated this week
- Accelerated minigrid environments with JAXâ128Updated 6 months ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ225Updated 3 weeks ago
- Simple single-file baselines for Q-Learning in pure-GPU settingâ137Updated 2 months ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilitiesâ342Updated 2 weeks ago
- Deep reinforcement learning without experience replay, target networks, or batch updates.â212Updated 2 weeks ago
- Cost aware hyperparameter tuning algorithmâ140Updated 7 months ago
- Efficient baselines for autocurricula in JAX.â178Updated 5 months ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ685Updated last week
- Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmindâ362Updated this week
- â213Updated 2 months ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.â654Updated 2 years ago
- 1 million FPS multi-agent driving simulatorâ268Updated this week
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".â355Updated 8 months ago
- The Fastest Deep Reinforcement Learning Libraryâ746Updated last month
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.â125Updated this week
- TensorDict is a pytorch dedicated tensor container.â876Updated this week
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.â668Updated this week
- Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Labâ649Updated 2 weeks ago
- âī¸ Vectorized RL game environments in JAXâ439Updated this week
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.â800Updated 6 months ago
- An API conversion tool for popular external reinforcement learning environmentsâ151Updated last month
- â207Updated 7 months ago