/apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

Pinned issues

Can not use tensor cores

#221 opened by vaibhav0195

Closed8

Install Failure on GCP Deep Learning VM

#259 opened by glenn-jocher

Closed21

torch.cuda.amp > apex.amp

#818 opened by mcarilli

Open21

Issues