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

  • Provide high-level, pythonic APIs for accelerating analog quantum dynamics solvers based on the quantum many-body operators and density-matrix (or state-vector) formalism

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

Support

  • Supported Python versions: 3.10, 3.11, 3.12

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

Prerequisites

  • CUDA 11 & 12

  • cuStateVec 1.7.0+

  • cuTensorNet 2.6.0+

  • cuDensityMat >=0.0.5, <0.1.0

  • NumPy v1.21+

  • CuPy v13.0+

  • Optional dependencies:

Contents