Build and run Docker containers leveraging NVIDIA GPUs
Makefile 14.3k 1.9k
Deep Learning Examples
Python 8k 2.2k
An Open Source Machine Learning Framework for Everyone
C++ 373 64
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
NeMo: a toolkit for conversational AI
Spark RAPIDS plugin - accelerate Apache Spark with GPUs
AIStore: scalable storage for AI applications
Tools for building GPU clusters
A high-performance, scalable and efficient ShuffleManager plugin for Apache Spark, utilizing UCX communication layer
NVIDIA Federated Learning Application Runtime Environment
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT
RAPIDS Accelerator JNI For Apache Spark
Run cloud native workloads on NVIDIA GPUs