pytorch-labs / LeanRL
LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
☆524Updated 4 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☆841Updated 6 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆294Updated 3 weeks ago
- Multi-Agent Reinforcement Learning with JAX☆541Updated 2 weeks ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆293Updated this week
- Distributed Reinforcement Learning accelerated by Lightning Fabric☆355Updated this week
- RL Environments in JAX 🌍☆721Updated 8 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆423Updated 2 weeks ago