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 56 results

Kaldi
NVIDIA
Kaldi is an open-source software framework for speech processing.
Container
Text Classification with BERT and NeMo. This NeMo application trains text classification models using single-GPU or multi-GPU. We log performance metrics and visualize them with TensorBoard. We show how to do inference with NeMo, and we visualize BERT embeddings before and after fine-tuning.
Container
DeepPavlov
DeepPavlov
DeepPavlov is an open-source conversational AI library built on TensorFlow and Keras. DeepPavlov is designed for development of production ready chatbots and complex conversational systems, research in the area of NLP and, particularly, of dialog systems.
Container
Jupyter Notebooks for BERT Pre-training, Fine-Tuning and Inference profiling and optimization via TensorFlow, AMP, XLA, DLProf, TF-TRT and Triton.
Container
345M parameter GPT generative Megatron model
Model
BERT Large Model trained with NeMo on uncased Wikipedia and Bookcorpus on a sequence length 512.
Model
BERT Base Model trained on uncased Wikipedia and BookCorpus dataset on a sequence length of 512.
Model
Megatron pretrained on cased biomedical dataset PubMed with 345 million parameters.
Model
BioMegatron 345M uncased model for Question Answering finetuned with NeMo on SQuAD v1.1 dataset.
Model
BERT Base Uncased model for Question Answering finetuned with NeMo on SQuAD v1.1 dataset.
Model
Model
BERT Distilled 4L-288D PyTorch checkpoint distilled on SQuAD v1.1 dataset using AMP
Model
BERT Base Model trained with NeMo on cased Wikipedia and BookCorpus on a sequence length of 512.
Model
BERT Large PyTorch checkpoint finetuned on GLUE/SST-2 dataset using AMP
Model
Checkpoint of TRADE model for dialogue state tracking trained on MultiWOZ 2.1 dataset using NeMo.
Model
Checkpoint of TRADE model for dialogue state tracking trained on MultiWOZ 2.0 dataset using NeMo.
Model
Bert Large TensorFlow2 checkpoint finetuned on Squad1.1 using seqLen=384
Model
BERT Large PaddlePaddle checkpoint pretrained with LAMB optimizer using AMP
Model
BERT Distilled 6L-768D PyTorch checkpoint distilled on SQuAD v1.1 dataset using AMP
Model
BERT Base PyTorch checkpoint pretrained with LAMB optimizer using AMP
Model
BERT Distilled 6L-768D PyTorch Phase2 checkpoint pretrained using 67K steps on seqLen128 and 6k steps on seqLen512
Model
BERT Distilled 4L-288D PyTorch checkpoint distilled on pretraining dataset using AMP
Model
BART PyT checkpoint (Summarization, XSum)
NVIDIA Deep Learning Examples
BART PyT checkpoint for summarization on XSum dataset
Model
Bert Large TensorFlow2 checkpoint pretrained using AMP and LAMB optimizer
Model
1-24 of 56 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.