Tsung-Hsien Lee
- A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scale
2023/09/12 by Hao-Jun Michael Shi, Tsung-Hsien Lee, Shi, Hao-Jun Michael +13 · 15 citations
Computer Science · #Advanced Neural Network Applications #Stochastic Gradient Optimization Techniques #Neural Networks and Applications