Paul Janson
Paul Janson
Home
Publications
Contact
Light
Dark
Automatic
learned optimizers
PyLO: Towards Accessible Learned Optimizers in PyTorch
A PyTorch library that makes learned optimizers accessible to the broader ML community with CUDA-accelerated implementations with substantial speedups (5x improvement for ViT training) and integrates seamlessly with existing PyTorch workflows, enabling practical application of learned optimization to real-world large-scale tasks.
Paul Janson
,
Benjamin Therien
,
Quentin Anthony
,
Xialong Huang
,
Abhinav Moudgil
,
Eugene Belilovsky
PDF
Cite
Code
Post
Cite
×