NVIDIA

Security scanner for AI agent skills. Detect vulnerabilities, malicious patterns, security risks, prompt injection, data exfiltration, and supply-chain risks in Claude Code, Codex, and MCP skills...

Run agents like Hermes, LangChain Deep Agents, and OpenClaw more securely inside NVIDIA OpenShell with managed inference

OpenShell is the safe, private runtime for autonomous AI agents.

Ongoing research training transformer models at scale

Build and run Docker containers leveraging NVIDIA GPUs

NVIDIA Linux open GPU kernel module source

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....

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.

NVIDIA Cosmos is an open platform of world models, datasets, and tools that enables developers to build Physical AI for robots, autonomous vehicles, smart infrastructure, and more.

PersonaPlex code.

Auto Ontology is a Python framework for building semantic, graph-based reasoning layers over data. It enables deterministic query generation, better context retrieval, and more explainable AI for...

Style transfer, deep learning, feature transform

CUDA Templates and Python DSLs for High-Performance Linear Algebra

the LLM vulnerability scanner

cuDF - GPU DataFrame Library

Samples for CUDA Developers which demonstrates features in CUDA Toolkit

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.

NVIDIA Isaac GR00T N1.7 - A Foundation Model for Generalist Robots.

A Python framework for GPU-accelerated simulation, robotics, and machine learning.

Synthesizing and manipulating 2048x1024 images with conditional GANs

Transformer related optimization, including BERT, GPT

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

A unified library of SOTA model optimization techniques like quantization, distillation, pruning, neural architecture search, speculative decoding, etc. It compresses deep learning models for...

NVIDIA cuML: GPU-Accelerated Machine Learning

Tacotron 2 - PyTorch implementation with faster-than-realtime inference

Optimized primitives for collective multi-GPU communication

[ARCHIVED] The C++ parallel algorithms library. See https://github.com/NVIDIA/cccl

Build and run containers leveraging NVIDIA GPUs

Generative AI reference workflows optimized for accelerated infrastructure and microservice architecture.

Deep Learning GPU Training System

NVIDIA's CUDA platform for Rust. Host runtime crates plus Tile (cutile-rs) and SIMT (cuda-oxide) kernel programming models in idiomatic Rust.

NVIDIA device plugin for Kubernetes

Agent Skills for NVIDIA products — install into Claude Code, Codex, and other coding agents to run Physical AI, robotics, simulation, CUDA, and RAG workflows end to end.

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better...

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

CUDA Python: Performance meets Productivity

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

A developer reference project for creating Retrieval Augmented Generation (RAG) chatbots on Windows using TensorRT-LLM

NeMo Retriever Library is a scalable, performance-oriented document content and metadata extraction microservice. NeMo Retriever Library uses specialized NVIDIA NIM microservices to find,...

NVIDIA GPU Operator creates, configures, and manages GPUs in Kubernetes

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

The NVIDIA NeMo Agent toolkit is an open-source library for efficiently connecting and optimizing teams of AI agents.

CUDA Core Compute Libraries

CUDA Library Samples

`std::execution`, the standard C++ framework for asynchronous and parallel programming.

A Flow-based Generative Network for Speech Synthesis

[ARCHIVED] The C++ Standard Library for your entire system. See https://github.com/NVIDIA/cccl

NVIDIA curated collection of educational resources related to general purpose GPU programming.

cuTile is a programming model for writing parallel kernels for NVIDIA GPUs

TensorRT Extension for Stable Diffusion Web UI

AIStore: scalable storage for AI applications

NVIDIA hardware telemetry for Prometheus.

[ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl

Nvidia Semantic Segmentation monorepo

A suite of image and video neural tokenizers

Router that virtually distributes inference across connected devices in the home.

Deep learning for recommender systems

NCCL Tests

Simple samples for TensorRT programming

NVIDIA DLSS is a new and improved deep learning neural network that boosts frame rates and generates beautiful, sharp images for your games

This repo contains the source code for RULER: What’s the Real Context Size of Your Long-Context Language Models?

Toolkit for efficient experimentation with Speech Recognition, Text2Speech and NLP

Tools for building GPU clusters

Collection of step-by-step playbooks for setting up AI/ML workloads on NVIDIA DGX Spark devices with Blackwell architecture.

An efficient C++20 GPU numerical computing library with Python-like syntax

A fast GPU memory copy library based on NVIDIA GPUDirect RDMA technology

Set of Python bindings to C++ libraries which provides full HW acceleration for video decoding, encoding and GPU-accelerated color space and pixel format conversions

Documentation of NVIDIA chip/hardware interfaces

NVIDIA’s implementation of RTX ray-tracing in Quake II

A New Padding Scheme: Partial Convolution based Padding

LLM KV cache compression made easy

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.

Official PyTorch implementation of BigVGAN (ICLR 2023)

PyTorch implementation of Audio Flamingo: Series of Advanced Audio Understanding Language Models

C++ and Python support for the CUDA Quantum programming model for heterogeneous quantum-classical workflows

An Open Source Machine Learning Framework for Everyone

NVIDIA container runtime library

Official Codebase for "DreamDojo: A Generalist Robot World Model from Large-Scale Human Videos" (ICML 2026)

NVIDIA container runtime

A simple yet powerful tool to turn traditional container/OS images into unprivileged sandboxes.

GPU accelerated decision optimization

A Python library that enables the use of Jetson's GPIOs

Tools for monitoring NVIDIA GPUs on Linux

Unsupervised Language Modeling at scale for robust sentiment classification

CUDA Tile IR is an MLIR-based intermediate representation and compiler infrastructure for CUDA kernel optimization, focusing on tile-based computation patterns and optimizations targeting NVIDIA...

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing...

NVIDIA Federated Learning Application Runtime Environment

NVIDIA cuDF for Apache Spark plugin - accelerate Apache Spark with GPUs

cuDNN Frontend is NVIDIA's modern, open-source entry point to the cuDNN library and a growing collection of high-performance open-source kernels.

CUDA Kernel Benchmarking Library

Examples demonstrating available options to program multiple GPUs in a single node or a cluster

Fast and accurate object detection with end-to-end GPU optimization

Flowtron is an auto-regressive flow-based generative network for text to speech synthesis with control over speech variation and style transfer

cuVS - a library for vector search and clustering on the GPU

Mellotron: a multispeaker voice synthesis model based on Tacotron 2 GST that can make a voice emote and sing without emotive or singing training data

Scalable toolkit for efficient model alignment

Helpful kernel tutorials, examples and SKILLs for tile-based GPU programming

NVIDIA Data Center GPU Manager (DCGM) is a project for gathering telemetry and measuring the health of NVIDIA GPUs