prodigy
The Prodigy optimizer and its variants for training neural networks.
Research Scientist
The Prodigy optimizer and its variants for training neural networks.
Benchmarking optimization methods on convex problems.
RR, SO, IG and SGD with logistic regression loss.
Code used for paper on Asynchronous SGD published at NeurIPS 2022
Proximal and Federated Random Reshuffling
Stochastic optimizers in pytorch
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
NanoGPT (124M) in 90 seconds