nvidia-cutlass-dsl-libs-cu12

What it is and what it does

nvidia-cutlass-dsl-libs-cu12 is NVIDIA's Python interface to CUTLASS 4.x, a library for writing optimized CUDA kernels without deep C++ expertise. It exposes CuTe DSL, a low-level programming model that maps directly to CuTe C++ abstractions—layouts, tensors, hardware atoms, and thread/data hierarchy—enabling developers to target NVIDIA's Tensor Cores on modern GPUs (Ampere, Hopper, Blackwell) with full control over kernel structure.

The package is currently in public beta (graduating by summer 2026) and is designed to flatten the GPU programming learning curve, accelerate kernel prototyping, and integrate natively with deep-learning frameworks. It requires CUDA 12, runs on Linux (x86_64 and aarch64), and depends on numpy, typing-extensions, cuda-python, protobuf, and other NVIDIA runtime libraries.

Use it for:

• Rapidly prototype and optimize matrix-multiply kernels for Tensor Cores without writing C++ code.
• Integrate custom CUDA kernels directly into deep-learning framework workflows using Python.
• Learn GPU programming and CuTe concepts with a lower barrier to entry than C++ CUTLASS.
• Develop high-throughput linear algebra operations targeting Ampere, Hopper, or Blackwell GPUs.
• Experiment with kernel designs and performance tuning in a native Python environment.

Worth the install?

AI-flagged interpretation of the facts on this page — verify before relying

Provides a Python DSL for writing high-performance CUDA kernels using CUTLASS and CuTe concepts, targeting NVIDIA Tensor Cores on Ampere, Hopper, and Blackwell architectures.

Yes, if you have a compatible NVIDIA GPU (Ampere/Hopper/Blackwell), CUDA 12 environment, and need to write optimized CUDA kernels in Python. The active maintenance, strong upstream signal, and zero known vulnerabilities support adoption. However, verify the unclear license terms for your use case, and note that the package is in beta—expect potential API changes before summer 2026.

Install

nvidia-cutlass-dsl-libs-cu12 on PyPI

pip install nvidia-cutlass-dsl-libs-cu12

uv add nvidia-cutlass-dsl-libs-cu12

poetry add nvidia-cutlass-dsl-libs-cu12

Installing nvidia-cutlass-dsl-libs-cu12

Before you install

Medium install friction due to platform-specific wheels (x86_64 and aarch64 Linux only, Python 3.10–3.14) and seven runtime dependencies including cuda-python and nvidia-cuda-nvdisasm. Active maintenance with recent release (9 days old) and strong upstream repository signal (10250 stars).

License in practice

License treatment is unclear; no SPDX identifier or raw license text is available. Verify licensing terms before use in proprietary or commercial contexts.

Quickstart

pip install nvidia-cutlass-dsl-libs-cu12
import numpy
# Write kernels targeting Tensor Cores using CuTe DSL concepts

Requires CUDA 12, Linux (x86_64 or aarch64), Python ≥3.10, and NVIDIA GPU with Ampere, Hopper, or Blackwell architecture.

Verify before relying

• Specific API surface and programming model details beyond CuTe DSL concepts (layouts, tensors, hardware atoms).
• Performance benchmarks or comparative compile-time improvements claimed in the description.
• Beta stability guarantees and breaking-change policy before summer 2026 graduation.
• Integration patterns with deep-learning frameworks mentioned in description.

Package facts

Evidence: nvidia_cutlass_dsl_libs_cu12-4.7.0-cp310-cp310-manylinux_2_28_aarch64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp310-cp310-manylinux_2_28_x86_64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp311-cp311-manylinux_2_28_aarch64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp311-cp311-manylinux_2_28_x86_64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp312-cp312-manylinux_2_28_aarch64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp312-cp312-manylinux_2_28_x86_64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp313-cp313-manylinux_2_28_aarch64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp313-cp313-manylinux_2_28_x86_64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp314-cp314-manylinux_2_28_aarch64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp314-cp314-manylinux_2_28_x86_64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp314-cp314t-manylinux_2_28_aarch64.whl; nvidia_cutlass_dsl_libs_cu12-4.7.0-cp314-cp314t-manylinux_2_28_x86_64.whl

Tags

More Scientific/Engineering packages

Further reading