CUDA Python is the home for accessing NVIDIA's CUDA platform from Python.
This package provides type stubs (and docstrings) for python cuda bindings. Enables features like auto-completion and hovering documentation when working with cuda-python in an IDE (like Visual Studio Code).
pip install cuda-stubs