GPU-optimized AI, Machine Learning, & HPC Software

NVIDIA

cBottle is an diffusion model that generates atmospheric states at kilometer resolution using a cascaded diffusion architecture.

Model

NVIDIA Developer Program

NVIDIA

NVIDIA NIM for Text Embedding

Helm Chart for NeMo Retriever Text Embedding NIM

Helm Chart

NVIDIA Deep Learning Examples

ResNet50 v1.5 for PyTorch

With modified architecture and initialization this ResNet50 version gives ~0.5% better accuracy than original.

Resource

Mistral AI

Mistral-7B Chat Int4

The Mistral-7B-Instruct-v0.1 Large Language Model (LLM) is a instruct fine-tuned version of the Mistral-7B-v0.1 generative text model using a variety of publicly available conversation datasets.

Model

NVIDIA

FoundationPose

FoundationPose is a unified foundation model for 6D object pose estimation and tracking of objects.

Model

NVIDIA

DeepStream

DeepStream SDK delivers a complete streaming analytics toolkit for AI based video and image understanding and multi-sensor processing. (Resources/ TAR and * Debian)

Resource

NVIDIA

StyleGAN3 pretrained models

StyleGAN3 pretrained models for FFHQ, AFHQv2 and MetFaces datasets.

Model

NVIDIA AI Enterprise

NVIDIA

Kit SDK Linux (FB)

Toolkit for building native Omniverse applications and microservices.

Resource

NVIDIA AI Enterprise

NVIDIA

Maxine Windows Augmented Reality SDK

Maxine Augmented Reality SDK enables real-time modeling and tracking of human faces from video.

Resource

NVIDIA AI Enterprise

NVIDIA

Kit SDK Windows (FB)

Toolkit for building native Omniverse applications and microservices.

Resource

Knowledge Engineering Group (KEG) & Data Mining at Tsinghua University

ChatGLM3-6B Chat Int4

ChatGLM3-6B is the latest open-source model in the ChatGLM series. ChatGLM3-6B introduces the following features (1) More Powerful Base Model (2) More Comprehensive Function Support (3) More Comprehensive Open-source Series.

Model

NVIDIA Developer Program

NVIDIA

NVIDIA NIM for LLMs

Helm Chart for NVIDIA NIM for LLMs

Helm Chart

NVIDIA

Riva Skills Quick Start

Scripts and utilities for getting started with Riva Speech Skills

Resource

NVIDIA Developer Program

NVIDIA

DeepSeek-R1

NVIDIA NIM for GPU accelerated DeepSeek-R1 inference through OpenAI compatible APIs

Container

NVIDIA

PeopleNet

3 class object detection network to detect people in an image.

Model

NVIDIA

Riva Speech Skills

Riva Speech Skills Helm Chart

Helm Chart

NVIDIA

NeMo Framework Megatron Backend

NVIDIA NeMo™ framework Megatron backend supports pre-training, post-training, and reinforcement learning of LLMs and multi-modal generative AI models with state-of-the-art data processing, model training techniques, and flexible deployment options.

Container

NVIDIA

TensorRT

NVIDIA TensorRT is a C++ library that facilitates high-performance inference on NVIDIA graphics processing units (GPUs). TensorRT takes a trained network and produces a highly optimized runtime engine that performs inference for that network.

Container

Google

TensorFlow

TensorFlow is an open source platform for machine learning. It provides comprehensive tools and libraries in a flexible architecture allowing easy deployment across a variety of platforms and devices.

Container

NVIDIA

PyTorch

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

NVIDIA

NVIDIA Network Operator

NVIDIA Network Operator Helm Chart provides an easy way to install, configure and manage the lifecycle of NVIDIA Mellanox network operator.

Helm Chart

NVIDIA

cuda

Container registry for CUDA images

Container

NVIDIA

Triton Inference Server

Triton Inference Server is an open source software that lets teams deploy trained AI models from any framework, from local or cloud storage and on any GPU- or CPU-based infrastructure in the cloud, data center, or embedded devices.

Container

NVIDIA

GPU Operator

Deploy and Manage NVIDIA GPU resources in Kubernetes.

Helm Chart

Collection

Cosmos Transfer1 7B

NVIDIA

Cosmos World Foundation Models

Cosmos World Foundation Models: A family of highly performant pre-trained world foundation models purpose-built for generating physics-aware videos and world states for physical AI development.

815

Collection

NVIDIA

DeepStream SDK

NVIDIA DeepStream SDK enables developers to build accelerated pipelines for a wide range of use-cases such as IVA, retail, industrial inspection, and many more with minimal development effort.

2273

Collection

NVIDIA

NeMo Microservices

This collection contains NeMo microservices that provide a comprehensive suite of features to build an end-to-end platform for fine-tuning, evaluating, and serving large language models (LLMs) on your Kubernetes cluster.

2314

Collection

NVIDIA AI Enterprise

NVIDIA

NVIDIA AI Enterprise Infra 6

Access the latest infrastructure software and tools of the NVIDIA AI Enterprise 6 release, exclusively available with your NVIDIA AI Enterprise subscription.

333

Collection

NVIDIA AI Enterprise

NVIDIA

Omniverse Kit (FB)

Feature Branches provide cutting-edge updates, incorporating the latest Kit Application Framework, Kit Extensions, and Omniverse SDKs. These branches are regulary updated and tailored for developers needing early access to innovations.

110

Collection

NVIDIA AI Enterprise

NVIDIA

Omniverse Kit App Streaming

551