LeanRL

Fast PyTorch RL with Torch.compile and CUDA Graphs. LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs. We're excited to announce that we've open-sourced LeanRL, a lightweight PyTorch reinforcement learning library that provides recipes for fast RL training using torch.compile and CUDA graphs.

Open source

704 stars · GitHubexternal siteArchived

Visit github.comexternal site