Build and run Docker containers leveraging NVIDIA GPUs
Makefile 13.7k 1.8k
Deep Learning Examples
Python 6.6k 2k
An Open Source Machine Learning Framework for Everyone
C++ 261 44
NeMo: a toolkit for conversational AI
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Facilitates searching, screening, and organizing large chemical databases
Spark RAPIDS plugin - accelerate Apache Spark with GPUs
NVIDIA GPU Operator creates/configures/manages GPUs atop Kubernetes
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT
Cooperative primitives for CUDA C++.
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.