a Pytorch extension with NVIDIA-maintained utilities to streamline mixed precision and distributed training.
There are no labels for this package.