a Pytorch extension with NVIDIA-maintained utilities to streamline mixed precision and distributed training.
conda install angloyna::nvidia-apex