An Evaluation of Memory Optimization Methods for Training Neural Networks

26 March 2023

Xiaoxuan Liu

Siddharth Jha

Papers citing "An Evaluation of Memory Optimization Methods for Training Neural Networks"

2 / 2 papers shown

Title
ZeRO-Offload: Democratizing Billion-Scale Model Training Jie Ren Samyam Rajbhandari Reza Yazdani Aminabadi Olatunji Ruwase Shuangyang Yang Minjia Zhang Dong Li Yuxiong He MoE 168 414 0 18 Jan 2021
Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism M. Shoeybi M. Patwary Raul Puri P. LeGresley Jared Casper Bryan Catanzaro MoE 245 1,821 0 17 Sep 2019