a Pytorch extension with NVIDIA-maintained utilities to streamline mixed precision and distributed training.