NVIDIA

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

Ongoing research training transformer models at scale

NVIDIA Linux open GPU kernel module source

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 before you install them.

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

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. TensorRT LLM also contains components to create Python and C++ runtimes that orchestrate the inference execution in a performant way.

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.

CUDA Templates and Python DSLs for High-Performance Linear Algebra

cuDF - GPU DataFrame Library

Samples for CUDA Developers which demonstrates features in CUDA Toolkit

the LLM vulnerability scanner

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

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

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.

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

NVIDIA cuML: GPU-Accelerated Machine Learning

Optimized primitives for collective multi-GPU communication

Build and run containers leveraging NVIDIA GPUs

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

NVIDIA device plugin for Kubernetes

A unified library of SOTA model optimization techniques like quantization, distillation, pruning, neural architecture search, speculative decoding, etc. It compresses deep learning models for downstream deployment frameworks like TensorRT-LLM, TensorRT, vLLM, etc. to optimize inference speed.

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 performance with lower memory utilization in both training and inference.

CUDA Python: Performance meets Productivity

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

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

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.

NeMo Retriever Library is a scalable, performance-oriented document content and metadata extraction microservice. NeMo Retriever Library uses specialized NVIDIA NIM microservices to find, contextualize, and extract text, tables, charts and images that you can use in downstream generative applications.

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

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

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.

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

TensorRT Extension for Stable Diffusion Web UI

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

AIStore: scalable storage for AI applications

NVIDIA GPU metrics exporter for Prometheus leveraging DCGM

Simple samples for TensorRT programming

NCCL Tests

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

Tools for building GPU clusters

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

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

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

Documentation of NVIDIA chip/hardware interfaces

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

Official PyTorch implementation of BigVGAN (ICLR 2023)

LLM KV cache compression made easy

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

NVIDIA container runtime library

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

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

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

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

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

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 high performance applications.

GPU accelerated decision optimization

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 tensor core units.

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

NVIDIA Federated Learning Application Runtime Environment

CUDA Kernel Benchmarking Library

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

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

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

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

A tool for bandwidth measurements on NVIDIA GPUs.

Differentiable signal processing on the sphere for PyTorch

Distributed multigrid linear solver library on GPU

Deep Learning Experiment Management

DreamGen: Nvidia GEAR Lab's initiative to solve the robotics data problem using world models

NVIDIA Math Libraries for the Python Ecosystem

NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process communication and coordination overheads by allowing programmers to perform one-sided communication from within CUDA kernels and on CUDA streams.

A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).

The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resources in your applications.

SOMA BVH to humanoid robot motion retargeting library built with Newton and NVIDIA Warp

The GL Vendor-Neutral Dispatch library

NVIDIA Material Definition Language SDK

High-performance, light-weight C++ LLM and VLM Inference Software for Physical AI

HPC Container Maker