CUDA Python: Performance meets Productivity
☆3,173Updated this week
Alternatives and similar repositories for cuda-python
Users that are interested in cuda-python are comparing it to the libraries listed below
Sorting:
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,315Updated this week
- Development repository for the Triton language and compiler☆18,460Updated this week
- NumPy & SciPy for GPU☆10,804Updated this week
- A Python framework for accelerated simulation, data generation and spatial computing.☆6,248Updated this week
- CUDA Core Compute Libraries☆2,182Updated this week
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on H…☆3,170Feb 21, 2026Updated last week
- TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizat…☆12,938Updated this week
- A PyTorch native platform for training generative AI models☆5,084Updated this week
- Fast and memory-efficient exact attention☆22,361Updated this week
- FlashInfer: Kernel Library for LLM Serving☆5,009Updated this week
- The CUDA target for Numba☆259Updated this week
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,947Updated this week
- SGLang is a high-performance serving framework for large language models and multimodal models.☆23,658Updated this week
- CUDA Library Samples☆2,324Updated this week
- PyTorch native quantization and sparsity for training and inference☆2,696Updated this week
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆5,236Feb 20, 2026Updated last week
- NVIDIA Math Libraries for the Python Ecosystem☆552Jan 16, 2026Updated last month
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,261Aug 28, 2025Updated 6 months ago
- Samples for CUDA Developers which demonstrates features in CUDA Toolkit☆8,870Jan 6, 2026Updated last month
- A Datacenter Scale Distributed Inference Serving Framework☆6,117Updated this week
- Distributed Compiler based on Triton for Parallel Systems☆1,361Feb 13, 2026Updated 2 weeks ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,723Updated this week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,940Updated this week
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,141Feb 19, 2026Updated last week
- CUDA Kernel Benchmarking Library☆820Updated this week
- A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Auto…☆16,807Updated this week
- Material for gpu-mode lectures☆5,773Feb 1, 2026Updated 3 weeks ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆71,234Updated this week
- CUDA integration for Python, plus shiny features☆2,025Feb 18, 2026Updated last week
- Lightning fast C++/CUDA neural network framework☆4,418Dec 14, 2025Updated 2 months ago
- Tile primitives for speedy kernels☆3,183Updated this week
- Ongoing research training transformer models at scale☆15,242Updated this week
- Efficient Triton Kernels for LLM Training☆6,162Updated this week
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆6,185Feb 3, 2026Updated 3 weeks ago
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,648Updated this week
- Transformer related optimization, including BERT, GPT☆6,394Mar 27, 2024Updated last year
- NumPy and SciPy on Multi-Node Multi-GPU systems☆967Feb 20, 2026Updated last week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,706Jan 12, 2026Updated last month
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆379Updated this week