AdamW-Triton-PyTorch. Can AdamW written in Triton be as performant as fused CUDA impl?

github.com/lessw2020/AdamW-Triton-PyTorch

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.