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

ASR + BERT QA interactive chatbot demo for Jetson
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
Jupyter Notebook example for Question Answering with BERT for TensorFlow
Resource
Base environment used in the NVIDIA NeMo projects of the NVIDIA Deep Learning Institute (DLI) course, "Building Transformer-Based Natural Language Processing Applications". This container also includes a "Next Steps" project.
Container
Domain classification of the query for weather chat bot.
Model
Fine-tune a pre-trained BERT model with the SQuAD dataset, optimize for inference using TensorRT and deploy with Triton Inference Server on Google Cloud AI Platform using Custom Containers
Resource
BERT Large Uncased trained on English Wikipedia and BookCorpus
Model
Megatron pretrained on uncased biomedical dataset PubMed with 345 million parameters.
Model
BERT Base Uncased trained on English Wikipedia and BookCorpus
Model
End to End workflow for question answering starting with training in TAO Toolkit and deployment using Riva.
Resource
For each word in the input text, the model: 1) predicts a punctuation mark that should follow the word (if any), the model supports commas, periods and question marks) and 2) predicts if the word should be capitalized or not.
Model
BioBERT-large cased model checkpoint for NeMo.
Model
End to End sample workflow for Text Classification starting with training in TAO Toolkit and deployment using Riva.
Resource
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
Punctuation and Capitalization model with BERT
Model
For each word in the input text, the model: 1) predicts a punctuation mark that should follow the word (if any), the model supports commas, periods and question marks) and 2) predicts if the word should be capitalized or not.
Model
BioMegatron 345M uncased model for Question Answering finetuned with NeMo on SQuAD v1.1 dataset.
Model
Named Entity Recognition model with BERT
Model
BERT Base Uncased model for Question Answering finetuned with NeMo on SQuAD v1.1 dataset.
Model
Intent and Slot classification of the queries for the misty bot with DistilBert model trained on weather, smalltalk and POI (places of interest) data.
Model
1-24 of 66 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.