3 ms·Distributed Shampoo Code Release: ResNet-50 in 1729 steps, 284 secs on TPUs1 points by moultano 6y ago