NVIDIA / NVIDIA/cuda-python

[FEA]: Type stubs / .pyi files for cuda.bindings and cuda.core

Open
#1,963 11 comments 0 reactions 1 assignee View on GitHub

@mdboom is already working on this.

Since May 7, 2026.

cuda.bindings enhancement P1 packaging
Dominant language
Cython
Stars
3.4k
Forks
329
Avg merge
1d 23h
Merged PRs (30d)
116

Description

Is your feature request related to a problem? Please describe.

The cuda.bindings module is implemented as a C extension, which means type checkers (mypy, pyright) and IDEs (VS Code, PyCharm) cannot introspect the API for autocompletion, inline documentation, or static type checking without explicit type information.

Describe the solution you'd like

Ship PEP 561-compliant type stubs (.pyi files) for cuda.bindings and other public modules. Ideally these would be:

  • Auto-generated alongside the bindings (to stay in sync with the C API)
  • Distributed inline with the package (with a py.typed marker)

Additional context

This has been requested by users in internal Slack channels. Type stubs are on the roadmap for the cuda.bindings track.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.