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.

오픈소스

704 스타 · GitHub외부 사이트보관됨

github.com 방문외부 사이트