cuQuantum Python: Python APIs for NVIDIA cuQuantum SDKΒΆ

Welcome to the cuQuantum Python documentation!

NVIDIA cuQuantum Python provides Python bindings and high-level object-oriented models for accessing the full functionalities of NVIDIA cuQuantum SDK from Python.

Available at https://github.com/NVIDIA/cuQuantum

Key Features

  • Provide 1:1 Python wrappers for all cuQuantum C APIs

  • Provide high-level, pythonic objects for creating and manipulating tensor networks

    • Interoperable with both CPU (NumPy, PyTorch) and GPU (CuPy, PyTorch) arrays

  • Provide a converter for Cirq and Qiskit users to map quantum circuits objects to tensor network contractions

  • Open sourced (under the BSD-3-Clause license) following the community practice for easy access

Support

  • Supported Python versions: 3.9, 3.10, 3.11, 3.12

  • Supported installation methods: conda install, pip install, or build from source

Prerequisites

  • CUDA 11 & 12

  • cuStateVec 1.6.0+

  • cuTensorNet 2.4.0+

  • NumPy v1.21+

  • CuPy v13.0+

  • Optional dependencies: