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

github.com/lessw2020/AdamW-Triton-PyTorch

Problem

Audience

Updates

No founder updates yet. Check back soon.