← Search

Aashaka Shah

1 accepted papers

2021

Memory Optimization for Deep Networks

ICLR 2021spotlight

Deep learning is slowly, but steadily, hitting a memory bottleneck. While the tensor computation in top-of-the-line GPUs increased by $32\times$ over the last five years, the total available memory only grew by $2.5\times$. This prevents researchers from exploring larger architectures, as training l…