prodigy. The Prodigy optimizer and its variants for training neural networks.
468opt_methods. Benchmarking optimization methods on convex problems.
36random_reshuffling. RR, SO, IG and SGD with logistic regression loss.
9asynchronous_sgd. Code used for paper on Asynchronous SGD published at NeurIPS 2022
8global-newton. Jupyter Notebook
5rr_prox_fed. Proximal and Federated Random Reshuffling
4pytorch_optimizers. Stochastic optimizers in pytorch
2Plume_labs. Jupyter Notebook
1proxyprox. Python
1examples. A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
1dave_rpg. Jupyter Notebook
1