Build and run Docker containers leveraging NVIDIA GPUs
Makefile 14.2k 1.8k
Deep Learning Examples
Jupyter Notebook 7.6k 2.2k
An Open Source Machine Learning Framework for Everyone
C++ 340 58
NeMo: a toolkit for conversational AI
Spark RAPIDS plugin - accelerate Apache Spark with GPUs
NVIDIA Federated Learning Application Runtime Environment
Optimized primitives for collective multi-GPU communication
TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Tools for building GPU clusters
RAPIDS Accelerator JNI For Apache Spark
The C++ Standard Library for your entire system.
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.