Cuda Kernels Python Packages

nvidia-cudnn-frontend

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

CUDA Core Compute Libraries

analysisg

A project which aims to abstract the analysis pipeline of a Particle Physics Analysis at ATLAS.

cuda-bench

CUDA Kernel Benchmarking Library

fast-trimul

A drop-in, hardware-agnostic library for Fused Triangle Multiplicative Updates across AlphaFold3-family models, powered by CUTLASS CuTe kernels.

serron

Mathematical Morphology module for PyTorch (with CUDA support), providing differentiable operators and learnable network layers.

pygameshader

2D Game texture special effects

kernel-tuner

Kernel Tuner

gridbook

Out-of-tree vLLM plugin and open format spec for NVFP4-CB / FP8-CB product-codebook weights — 2-6 bit-per-weight LLM quantization served on native Blackwell tensor cores.

flash-rt

FlashRT is a high-performance realtime inference engine for small-batch, latency-sensitive AI workloads. The flagship integration is production VLA control for Pi0, Pi0.5, GROOT N1.6, and Pi0-FAST. Also support llm e.g, qwen3.6-27B

event-analysis

Library for Event Synchronization and Event Coincidence Analysis

cutile-gpt

A complete GPT built entirely on NVIDIA cuTile: declarative GPU kernels in ~20 lines instead of ~150 hand-tuned CUDA, matching PyTorch speed in a ~10MB footprint

message-passing-nn

Pytorch implementation of a message passing neural network with RNN sub-units

cuda-engine

Plain English + a PyTorch reference → a verified, benchmarked CUDA kernel you can pip install and torch.compile.

dire-rapids

DiRe accelerated by PyTorch, PyKeOps and cuVS

gpuci

GPU CI/CD tool that tests CUDA kernels across multiple GPUs in parallel - Part of RightNow

graphem-rapids

Graph embedding for influence maximization in networks

gimbal-engine

GPU-accelerated video stabilization with a deep-learning homography model and a custom fused CUDA autograd kernel

cuda-parallel

CUDA Core Compute Libraries

cuda-cooperative

CUDA Core Compute Libraries