FeaturedDistributed training, from scratch
Small-LLM-RLHF
3D-parallel RLHF training for small language models, written from the ground up on PyTorch and Triton — no training framework underneath it.
Hand-rolled tensor, pipeline (1F1B), and data parallelism with ZeRO-1; custom Triton FlashAttention-2 kernels; a PagedAttention inference engine with its own block manager; and an async PPO loop that overlaps rollout generation on one GPU set with backprop on another.
- PyTorch
- Triton
- CUDA
- RLHF
- 3D Parallelism






