jettify / pytorch-optimizer

torch-optimizer -- collection of optimizers for Pytorch
Apache License 2.0
3.04k stars 299 forks source link

VeLO: Training Versatile Learned Optimizers by Scaling Up #479

Open hoangmit opened 2 years ago

hoangmit commented 2 years ago

Do we have any pytorch port of these learned optimizers ?

Link: https://arxiv.org/abs/2211.09760v1

AtomicityZ commented 1 year ago

Paper is very well written

Here is the repo for future reference.

Hope to see it soon here

drscotthawley commented 9 months ago

Just sharing Here's a version that was in development up to a year ago: https://github.com/janEbert/PyTorch-VeLO Haven't tried it yet and not sure if it (still) works.