NVIDIA/cuda-python

Health v2 · maintenance only

Adoption not part of the score

3359 stars · 328 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

NVIDIA's official Python package for accessing the CUDA platform, providing Pythonic APIs and low-level bindings to CUDA C APIs, parallel algorithms, JIT kernel compilation, and math libraries. It is an umbrella project comprising components like cuda.core, cuda.bindings, numba.cuda, cuda.tile, and nvmath-python.

Use cases

• write CUDA kernels in Python
• call CUDA C APIs from Python
• run parallel sort and reduce on the GPU
• compile Python functions to GPU code with Numba
• access cuBLAS and other NVIDIA math libraries from Python
• profile CUDA Python applications
• locate installed CUDA libraries in a Python environment

When to choose

• you need low-level control over CUDA from Python
• you want to write GPU kernels without leaving Python
• you need Python bindings for NVIDIA math or communication libraries
• you are building GPU-accelerated Python libraries or frameworks

When to avoid

• you only need high-level deep learning frameworks like PyTorch or TensorFlow
• you have no NVIDIA GPU or CUDA-capable hardware
• you need a pure-CPU numerical computing solution

Facets

library · maturity active

gpu-computing sdk compiler math concurrency gpu-computing developer-tools machine-learning data-science performance python windows cuda nvidia python-bindings gpu-kernels numba jit-compilation parallel-computing cython linux macos gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="NVIDIA/cuda-python")