nvidia-cutlass-dsl

What it is and what it does

nvidia-cutlass-dsl is NVIDIA's Python domain-specific language for writing optimized CUDA kernels that run on modern GPU Tensor Cores. It exposes core CuTe concepts—layouts, tensors, hardware atoms, and thread/data hierarchy control—directly in Python, eliminating the need to write C++ glue code or possess deep GPU programming expertise. The package targets Ampere, Hopper, and Blackwell architectures and is designed to accelerate matrix multiply and linear algebra operations.

The package is currently in public beta and depends on two runtime libraries (nvidia-cutlass-dsl-libs-base and nvidia-cutlass-dsl-libs-cu12) that provide the underlying compiled components. It supports Python 3.10 through 3.14 on Linux with CUDA 12 or 13, and installs as a pure wheel with low friction. The project is actively maintained by NVIDIA with recent commits and is positioned as a tool for students, researchers, and performance engineers to prototype and deploy GPU kernels.

Use it for:

• Rapidly prototype and optimize matrix multiplication kernels for deep learning workloads without writing C++ code
• Develop custom linear algebra operations targeting Tensor Cores on modern NVIDIA GPUs
• Integrate optimized GPU kernels directly into Python deep learning frameworks
• Learn GPU programming and Tensor Core concepts with a lower barrier to entry than C++ CUTLASS
• Benchmark and compare different kernel designs for high-throughput tensor operations

Worth the install?

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

Provides a Python native interface for writing high-performance CUDA kernels targeting NVIDIA Tensor Cores on Ampere, Hopper, and Blackwell architectures using CuTe DSL concepts without requiring C++ expertise.

Yes, if you need to write optimized CUDA kernels in Python and can tolerate beta-stage software. The low install friction, active maintenance, and clear focus on reducing GPU programming complexity make it valuable for researchers and performance engineers. However, verify NVIDIA's proprietary license terms for your use case, and be aware that the API may change before the summer 2026 beta graduation.

Install

nvidia-cutlass-dsl on PyPI

pip install nvidia-cutlass-dsl

uv add nvidia-cutlass-dsl

poetry add nvidia-cutlass-dsl

Installing nvidia-cutlass-dsl

Before you install

Low install friction with a pure-wheel distribution. The package is actively maintained with recent commits and is in public beta status, though it remains under active development.

License in practice

Licensed under an unclear proprietary license with no SPDX identifier published. You should review NVIDIA's licensing terms directly before committing to production use.

Quickstart

pip install nvidia-cutlass-dsl

# Import and use the package with its runtime dependencies nvidia-cutlass-dsl-libs-base and nvidia-cutlass-dsl-libs-cu12

Requires Linux, CUDA 12 or 13, Python 3.10 or later, and NVIDIA GPU with Ampere, Hopper, or Blackwell architecture to execute kernels.

Verify before relying

• Whether the beta status implies breaking API changes before summer 2026 graduation
• What specific performance guarantees or benchmarks are documented for different kernel types
• Whether the two runtime library dependencies can coexist or require selection
• What the actual import path and public API surface are for the package

Package facts

Evidence: nvidia_cutlass_dsl-4.7.0-py3-none-any.whl

Tags

More Scientific/Engineering packages

Further reading