NVIDIA / cuda-pythonLinks
CUDA Python: Performance meets Productivity
☆3,156Updated 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,903Updated this week
- PyTorch native quantization and sparsity for training and inference☆2,657Updated last 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,132Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,956Updated this week
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,226Updated this week
- CUDA Library Samples☆2,306Updated 2 weeks ago
- A unified library of SOTA model optimization techniques like quantization, pruning, distillation, speculative decoding, etc. It compresse…☆1,925Updated this week
- CUDA Core Compute Libraries☆2,162Updated this week
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆4,863Updated last week
- Tile primitives for speedy kernels☆3,120Updated this week
- PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri…☆1,439Updated this week
- SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, …☆2,577Updated last week
- A Datacenter Scale Distributed Inference Serving Framework☆6,019Updated this week
- FlashInfer: Kernel Library for LLM Serving☆4,853Updated this week
- A PyTorch native platform for training generative AI models☆5,023Updated last week
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆1,012Updated this week
- NVIDIA curated collection of educational resources related to general purpose GPU programming.☆1,146Updated this week
- NumPy and SciPy on Multi-Node Multi-GPU systems☆965Updated last week
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,104Updated last week
- NVIDIA Math Libraries for the Python Ecosystem☆544Updated 3 weeks ago
- Olive: Simplify ML Model Finetuning, Conversion, Quantization, and Optimization for CPUs, GPUs and NPUs.☆2,246Updated last week
- Optimized primitives for collective multi-GPU communication☆4,436Updated this week
- CUDA integration for Python, plus shiny features☆2,020Updated 3 weeks ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,525Updated this week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,072Updated last year
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆682Updated last week
- Efficient Triton Kernels for LLM Training☆6,123Updated this week
- CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.☆2,647Updated 2 weeks ago
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆864Updated 4 months ago
- NCCL Tests☆1,423Updated this week