LeanRL
Lightweight PyTorch library of reinforcement learning recipes that use torch.compile and CUDA graphs to train much faster than the CleanRL originals.
Open sourceCoding & Development
Visit website(opens in a new tab)
Lightweight PyTorch library of reinforcement learning recipes that use torch.compile and CUDA graphs to train much faster than the CleanRL originals.