Build and run Docker containers leveraging NVIDIA GPUs
Makefile 14.4k 1.9k
Deep Learning Examples
Python 8k 2.2k
An Open Source Machine Learning Framework for Everyone
C++ 376 65
The C++ parallel algorithms library.
NeMo: a toolkit for conversational AI
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
NVIDIA Federated Learning Application Runtime Environment
Spark RAPIDS plugin - accelerate Apache Spark with GPUs
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT