Distributed Shampoo Code Release: ResNet-50 in 1729 steps, 284 secs on TPUs | Heykuki News