Introducing Apex: PyTorch Extension with Tools to Realize the Power of Tensor Cores

Originally published at: https://developer.nvidia.com/blog/introducing-apex-pytorch-extension-with-tools-to-realize-the-power-of-tensor-cores/

Today at the Computer Vision and Pattern Recognition Conference in Salt Lake City, Utah, NVIDIA is kicking off the conference by demonstrating an early release of Apex, an open-source PyTorch extension that helps users maximize deep learning training performance on NVIDIA Volta GPUs. Inspired by state-of-the-art mixed precision training in translational networks, sentiment analysis, and…