Hao-Jun Michael Shi's starred repositories
algorithmic-efficiency
MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvements in both training algorithms and models.
PyTorch-LBFGS
A PyTorch implementation of L-BFGS.
large-batch-training
Code to reproduce some of the figures in the paper "On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima"
shampoo.pytorch
An implementation of shampoo