Nvidia Python Packages

nvidia-cutlass-dsl

CUDA Templates and Python DSLs for High-Performance Linear Algebra

flashinfer-python

FlashInfer: Kernel Library for LLM Serving

nvidia-cutlass-dsl-libs-base

CUDA Templates and Python DSLs for High-Performance Linear Algebra

flashinfer-cubin

FlashInfer: Kernel Library for LLM Serving

warp-lang

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

onnx-graphsurgeon

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

nvitop

An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.

tensorrt-cu12-bindings

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

tensorrt-cu12

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

nemoguardrails

NeMo Guardrails is an open-source toolkit for easily adding programmable guardrails to LLM-based conversational systems.

tensorrt-cu12-libs

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

tensorrt

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

polygraphy

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

pyopencl

OpenCL integration for Python, plus shiny features

cuml-cu12

cuML - RAPIDS Machine Learning Library

dstack

Vendor-agnostic orchestration for training, inference and agentic workloads across NVIDIA, AMD, TPU, and Tenstorrent on clouds, Kubernetes, and bare metal.

libcuml-cu12

cuML - RAPIDS Machine Learning Library

tensorrt-cu13-bindings

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

nvidia-nvimgcodec-cu12

A nvImageCodec library of GPU- and CPU- accelerated codecs featuring a unified interface

tensorrt-cu13-libs

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

tensorrt-cu13

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

cudensitymat-cu13

Home for cuQuantum Python & NVIDIA cuQuantum SDK C++ samples

cutensornet-cu13

Home for cuQuantum Python & NVIDIA cuQuantum SDK C++ samples

custatevec-cu13

Home for cuQuantum Python & NVIDIA cuQuantum SDK C++ samples