Build and run Docker containers leveraging NVIDIA GPUs
Makefile 14.5k 1.9k
Deep Learning Examples
Python 8.3k 2.3k
An Open Source Machine Learning Framework for Everyone
C++ 419 67
NeMo: a toolkit for conversational AI
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
Transformer related optimization, including BERT, GPT
RAPIDS Accelerator JNI For Apache Spark
Spark RAPIDS plugin - accelerate Apache Spark with GPUs
NVIDIA Federated Learning Application Runtime Environment
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Morpheus SDK
AIStore: scalable storage for AI applications