NVIDIA / cuda-python
CUDA Python: Performance meets Productivity
☆2,600Updated this week
Alternatives and similar repositories for cuda-python:
Users that are interested in cuda-python are comparing it to the libraries listed below
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…☆2,393Updated this week
- CUDA Core Compute Libraries☆1,626Updated this week
- Tile primitives for speedy kernels☆2,325Updated this week
- PyTorch native quantization and sparsity for training and inference☆2,015Updated this week
- Optimized primitives for collective multi-GPU communication☆3,690Updated last week
- FlashInfer: Kernel Library for LLM Serving☆2,788Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,135Updated this week
- NVIDIA Math Libraries for the Python Ecosystem☆310Updated last month
- Learn CUDA Programming, published by Packt☆1,136Updated last year
- A Datacenter Scale Distributed Inference Serving Framework☆3,931Updated this week
- A Python framework for accelerated simulation, data generation and spatial computing.☆5,034Updated this week
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆697Updated 2 months ago
- Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)☆756Updated 8 months ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,742Updated this week
- CUDA Library Samples☆1,912Updated last week
- NCCL Tests☆1,090Updated this week
- nvidia-modelopt is a unified library of state-of-the-art model optimization techniques like quantization, pruning, distillation, speculat…☆900Updated last week
- CUDA Templates for Linear Algebra Subroutines☆7,412Updated this week
- SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX R…☆2,387Updated last week
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆1,089Updated this week
- An Aspiring Drop-In Replacement for NumPy at Scale☆887Updated last week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆1,445Updated 2 months ago
- Puzzles for learning Triton☆1,614Updated 5 months ago
- common in-memory tensor structure☆984Updated 3 weeks ago
- NVIDIA curated collection of educational resources related to general purpose GPU programming.☆434Updated 2 weeks ago
- GPU programming related news and material links☆1,488Updated 4 months ago
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆381Updated this week
- 📚LeetCUDA: Modern CUDA Learn Notes with PyTorch for Beginners🐑, 200+ CUDA/Tensor Cores Kernels, HGEMM, FA-2 MMA etc.🔥☆3,896Updated last week
- CUDA integration for Python, plus shiny features☆1,929Updated 3 months ago
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,321Updated this week