Costa Huang

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

AllenAI's post-training codebase

JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

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