Lowering PyTorch's Memory Consumption for Selective Differentiation

15 April 2024

Papers citing "Lowering PyTorch's Memory Consumption for Selective Differentiation"

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 177 414 0 18 Jan 2021
Aggregated Residual Transformations for Deep Neural Networks Saining Xie Ross B. Girshick Piotr Dollár Z. Tu Kaiming He 297 10,220 0 16 Nov 2016