search

NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.42k stars 1.4k forks source link

Rui/dev fast ln #1726

Closed RuiWang1998 closed 1 year ago

RuiWang1998 commented 1 year ago

Moving MemEff to template