Repositories (34)
Build and run Docker containers leveraging NVIDIA GPUs
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
CUDA Templates for Linear Algebra Subroutines
NeMo: a toolkit for conversational AI
Synthesizing and manipulating 2048x1024 images with conditional GANs
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
The C++ parallel algorithms library.
Deep Learning GPU Training System
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.
The C++ Standard Library for your entire system.
NVIDIA device plugin for Kubernetes
Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
Deep learning for recommender systems
Toolkit for efficient experimentation with Speech Recognition, Text2Speech and NLP
Cooperative primitives for CUDA C++.
Fast and accurate object detection with end-to-end GPU optimization
CUDA Kernel Benchmarking Library
HPC Container Maker
A fast GPU memory copy library based on NVIDIA GPUDirect RDMA technology
A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).
GPU accelerated decision optimization
Providing reproducibility in deep learning frameworks
NVIDIA cuOpt examples for decision optimization
A tool for testing and validating container requirements against versioned manifests
A toolkit showing GPU's all-round capability in video processing
NVIDIA Dataset Utilities (NVDU)