← Search

Dehao Chen

2 accepted papers

2021

GShard: Scaling Giant Models with Conditional Computation and Automatic Sharding

ICLR 2021poster

Neural network scaling has been critical for improving the model quality in many real-world machine learning applications with vast amounts of training data and compute. Although this trend of scaling is affirmed to be a sure-fire approach for better model quality, there are challenges on the path s…

Cited by 1282SourcePDFScholar
2019

GPipe: Efficient Training of Giant Neural Networks using Pipeline Parallelism

NeurIPS 2019poster

Scaling up deep neural network capacity has been known as an effective approach to improving model quality for several different machine learning tasks. In many cases, increasing model capacity beyond the memory limit of a single accelerator has required developing special algorithms or infrastructu…