I built a PyTorch optimizer for large GPU workloads — looking for feedback[P]
This story is from 2026-10-27. It is preserved in the archive; the latest stories are on the live feed.
Hey, I've been working on a PyTorch optimizer called ApexW and figured I'd share it here to get some feedback from people who know more about this stuff than I do. The main goal is to reduce optimizer-state memory usage and see how much faster I can make optimizer steps on larger GPU workloads. I'v…
Read the full story at r/MachineLearning ↗
Timeline · 1 report
- 2026-10-27 12:11 · r/MachineLearning
I built a PyTorch optimizer for large GPU workloads — looking for feedback[P]