pytorch-labs/LeanRL

LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.

53 best points · 5 comments · 1 Hacker News post · 704 GitHub stars · View on GitHub

Hacker News discussions

• Show HN: LeanRL: Fast PyTorch RL with Torch.compile and CUDA Graphs (53 points, 5 comments, by vmoens, 2024-09-20)