NVIDIA and Facebook are delivering AI acceleration through its work on the Caffe2 deep learning framework. Caffe2 uses the latest NVIDIA Deep Learning SDK libraries — cuDNN, cuBLAS and NCCL — to deliver high-performance, multi-GPU accelerated training and inference.

Caffe2 is designed to be a fast, scalable and portable deep learning framework. It delivers near-linear scaling of deep learning training with 57x throughput acceleration on eight networked Facebook Big Basin AI servers with 64 NVIDIA Tesla P100 GPU accelerators. This means developers can train and iterate AI models faster than ever.

As part of the companies’ collaboration, the NVIDIA DGX-1 AI supercomputer will be the first AI system to offer Caffe2 within the optimized software stack for deep learning. Together, DGX-1 and Caffe2 deliver high performance and fast training. Caffe2 for DGX-1 will be available to customers via NVIDIA DGX-1 Container Registry.