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

PyTorch
NVIDIA
PyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels.
Container
TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs.
Container
PyTorch - AI Workbench Default Container
Container
GenMol is a masked diffusion model trained on molecular SAFE representations for fragment-based molecule generation, which can serve as a generalist model for various drug discovery tasks.
Container
A pre-trained model for volumetric (3D) segmentation of the spleen from CT image.
Model
Diffdock predicts the 3D structure of the interaction between a molecule and a protein.
Container
TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs.
Container
PyTorch is a GPU accelerated tensor computational framework with a Python front end. This container contains PyTorch and torchvision pre-installed in a Python 3 environment to get up & running quickly with PyTorch on Jetson.
Container
Get started on your AI journey quickly on Jetson. The Machine learning container contains TensorFlow, PyTorch, JupyterLab, and other popular ML and data science frameworks such as scikit-learn, scipy, and Pandas pre-installed in a Python environment.
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
The Merlin PyTorch container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with PyTorch, and serve the trained model on Triton Inference Server.
Container
PyTorch Production Branch October 2024 (PB 24h2) offers a 9-month lifecycle for API stability, with monthly patches for high and critical software vulnerabilities.
Container
Course environment for the Deep Learning Institute (DLI) course, "Getting Started with AI on Jetson Nano".
Container
NVIDIA Neural Reconstruction (NuRec) is a technology that converts recorded camera and lidar data into 3D scenes.
Container
This NIM serves as a demonstration of the potential of foundation models for early stage design evaluation in automotive aerodynamics.
Container
NVIDIA Neural Reconstruction Tools Container containing Aux Data Generation.
Container
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
PyTorch Production Branch October 2025 (PB 25h2) offers a 9-month lifecycle for API stability, with monthly patches for high and critical software vulnerabilities. This release includes Government Ready images for regulated environments.
Container
NIM Container for NV-Tesseract Time Series Analysis model.
Container
PyTorch Production Branch May 2025 (PB 25h1) offers a 9-month lifecycle for API stability, with monthly patches for high and critical software vulnerabilities. This release is a branch of PyTorch 25.03.
Container
PyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels.
Container
This container allows users to deploy NVTabular workflows and PyTorch models to Triton Inference server for production.
Container
PyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels.
Container
PyTorch Production Branch 6 offers a 9-month lifecycle for API stability, with monthly patches for high and critical software vulnerabilities. This release includes Government Ready images for regulated environments.
Container
1-24 of 84 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.