GPU-optimized AI, Machine Learning, & HPC Software

NVIDIA

This container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with PyTorch.

Container

NVIDIA

Merlin Training

This container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with HugeCTR.

Container

NVIDIA

Riva ASR English LM

Base English n-gram LM trained on LibriSpeech, Switchboard and Fisher

Model

University of Florida Health

GatorTron-S

GatorTron-S is a Megatron BERT model trained on pre-trained on synthetic clinical discharge summaries generated by SynGatorTron 5B NLG, a Megatron GPT-3 model trained on de-identified clinical free text at the University of Florida health system.

Model

NVIDIA

Riva ASR English(en-US) LM

Base English n-gram LM trained on LibriSpeech, Switchboard and Fisher

Model

Rendered.ai

TAO with Synthetic Data - Fast Demo

Jupyter notebook to train a Detectnet_v2 model with Rendered.ai user generated datasets.

Resource

NVIDIA

TAO Object Detection

This notebook shows an example usecase of DSSD object detection using Train Adapt Optimize (TAO) Launcher.

Resource