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

The post Introducing Apex: PyTorch Extension with Tools to Realize the Power of Tensor Cores appeared first on NVIDIA Developer News Center.