Skip to main content

Search

Search thousands of GPU-optimized Containers, pretrained Models, SDKs, and Helm charts—ready to accelerate AI, digital twins, and HPC from cloud to edge.
Accelerate custom generative AI app deployment using pre-built containers with optimized AI models.

Displaying 78 results

nemo-rl
NVIDIA
NVIDIA NeMo™ RL accelerates reinforcement learning post-training with high-performance GPU backends, offering scalable GRPO, DPO, SFT, and distillation for multimodal models from single-node experiments to enterprise-scale clusters.
Container
NVIDIA NeMo™ AutoModel accelerates LLM and VLM training and fine‑tuning with PyTorch DTensor‑native SPMD, day‑0 Hugging Face support, and optimized parallelism from single‑ to multi‑node scale.
Container
GPUNet-0 ImageNet pretrained weights
Model
GPUNet-P0 ImageNet pretrained weights
Model
GPUNet-D2 weights pretrained on ImageNet
Model
GPUNet-1 ImageNet pretrained weights
Model
GPUNet-2 ImageNet pretrained weights
Model
GPUNet-D1 weights pretrained on ImageNet
Model
GPUNet-P1 ImageNet pretrained weights
Model
HiFi-GAN PyT checkpoint (22kHz, AMP)
NVIDIA Deep Learning Examples
HiFi-GAN v1 PyTorch checkpoint trained on 8GPU with AMP on LJSpeech-1.1 (22kHz).
Model
HiFi-GAN v1 PyTorch checkpoint trained on 8GPU with AMP on LJSpeech-1.1 (22kHz), fine-tuned on FastPitch outputs.
Model
QuartzNet PyTorch checkpoint trained on LibriSpeech (test-other 10.41% WER)
Model
Fine-tuning and inference the Flowtron model which is an auto-regressive flow-based generative network for text to speech synthesis with control over speech variation and style transfer.
Resource
TFT PyT checkpoint (Base, AMP, Electricity)
NVIDIA Deep Learning Examples
TFT Base PyTorch checkpoint trained with AMP on Electricity dataset
Model
Mask R-CNN TensorFlow checkpoint (AMP)
NVIDIA Deep Learning Examples
Mask R-CNN TensorFlow checkpoint trained with AMP
Model
EfficientNets are a family of image classification models, which achieve state-of-the-art accuracy, being an order-of-magnitude smaller and faster.
Resource
BERT Distilled 4L-288D PyTorch checkpoint distilled on SQuAD v1.1 dataset using AMP
Model
ResNet50 checkpoint trained with AMP on ImageNet
Model
BERT Large PyTorch checkpoint finetuned on GLUE/SST-2 dataset using AMP
Model
TFT PyT checkpoint (Base, AMP, Traffic)
NVIDIA Deep Learning Examples
TFT Base PyTorch checkpoint trained with AMP on Traffic dataset
Model
FastPitch PyTorch checkpoint trained on LJSpeech-1.1
Model
BERT Distilled 6L-768D PyTorch checkpoint distilled on SQuAD v1.1 dataset using AMP
Model
Tacotron2 and WaveGlow PyTorch codebase
NVIDIA Deep Learning Examples
PyTorch codebase for training and using Tacotron2 and Waveglow models
Resource
BERT Distilled 6L-768D PyTorch Phase2 checkpoint pretrained using 67K steps on seqLen128 and 6k steps on seqLen512
Model
1-24 of 78 items

Modal Content

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.