NVIDIA / cuda-pythonLinks
CUDA Python: Performance meets Productivity
☆3,144Updated 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:
- cuTile is a programming model for writing parallel kernels for NVIDIA GPUs☆1,851Updated this week
- PyTorch native quantization and sparsity for training and inference☆2,645Updated this week
- NVIDIA curated collection of educational resources related to general purpose GPU programming.☆1,101Updated 2 weeks ago
- 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,116Updated this week
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,141Updated last week
- CUDA Library Samples☆2,288Updated last week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,923Updated this week
- CUDA Core Compute Libraries☆2,140Updated this week
- A unified library of SOTA model optimization techniques like quantization, pruning, distillation, speculative decoding, etc. It compresse…☆1,848Updated last week
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆4,806Updated this week
- A Datacenter Scale Distributed Inference Serving Framework☆5,961Updated this week
- Tile primitives for speedy kernels☆3,096Updated last week
- NumPy and SciPy on Multi-Node Multi-GPU systems☆965Updated this week
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆680Updated last month
- Optimized primitives for collective multi-GPU communication☆4,398Updated this week
- CUDA integration for Python, plus shiny features☆2,017Updated last week
- FlashInfer: Kernel Library for LLM Serving☆4,707Updated last week
- NVIDIA Math Libraries for the Python Ecosystem☆543Updated last week
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆859Updated 4 months ago
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆505Updated last week
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,928Updated this week
- Learn CUDA Programming, published by Packt☆1,227Updated 2 years ago
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,091Updated this week
- NCCL Tests☆1,406Updated last week
- SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, …☆2,574Updated this week
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,231Updated 4 months ago
- GPU programming related news and material links☆1,917Updated 4 months ago
- Development repository for the Triton language and compiler☆18,237Updated this week
- A PyTorch native platform for training generative AI models☆5,004Updated this week
- CUDA Kernel Benchmarking Library☆802Updated this week