cuquantum-python-jax 0.0.4__tar.gz → 0.0.7__tar.gz

{cuquantum_python_jax-0.0.4 → cuquantum_python_jax-0.0.7}/LICENSE

@@ -1,4 +1,4 @@
-Copyright (c) 2021-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 
 BSD-3-Clause
 

cuquantum_python_jax-0.0.7/PKG-INFO

@@ -0,0 +1,79 @@
+Metadata-Version: 2.4
+Name: cuquantum-python-jax
+Version: 0.0.7
+Summary: NVIDIA cuQuantum Python JAX
+Home-page: https://developer.nvidia.com/cuquantum-sdk
+Author: NVIDIA Corporation
+Author-email: cuquantum-python@nvidia.com
+License: BSD-3-Clause
+Project-URL: Bug Tracker, https://github.com/NVIDIA/cuQuantum/issues
+Project-URL: User Forum, https://github.com/NVIDIA/cuQuantum/discussions
+Project-URL: Documentation, https://docs.nvidia.com/cuda/cuquantum/latest/python/jax.html
+Project-URL: Source Code, https://github.com/NVIDIA/cuQuantum
+Keywords: cuda,nvidia,state vector,tensor network,high-performance computing,quantum computing
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Topic :: Education
+Classifier: Topic :: Scientific/Engineering
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Environment :: GPU :: NVIDIA CUDA
+Classifier: Environment :: GPU :: NVIDIA CUDA :: 12
+Classifier: Environment :: GPU :: NVIDIA CUDA :: 13
+Requires-Python: >=3.11
+Description-Content-Type: text/x-rst
+License-File: LICENSE
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: requires-python
+Dynamic: summary
+
+**********************
+cuQuantum Python JAX
+**********************
+
+cuQuantum Python JAX provides a JAX extension for cuQuantum Python. It exposes selected functionalities of
+cuQuantum SDK in a JAX-compatible way that enables JAX frameworks to directly interface with the exposed
+cuQuantum API. In the current release, cuQuantum JAX exposes JAX interfaces to the Operator Action API from
+the cuDensityMat library and the GF(2) sparse-dense matrix multiplication from the cuStabilizer library.
+
+Documentation
+=============
+
+For detailed guide, please refer to `cuQuantum Python JAX documentation <https://docs.nvidia.com/cuda/cuquantum/latest/python/jax.html>`_.
+
+Installation
+============
+
+.. code-block:: bash
+
+   pip install -v --no-cache-dir cuquantum-python-jax
+
+.. note::
+
+   This package is a meta package pointing to ``cuquantum-python-jax-cuXX``,
+   where XX is the CUDA major version (currently CUDA 12 & 13 are supported).
+   The meta package will attempt to infer and install the correct ``-cuXX`` wheel.
+   The auto-detection mechanism is not guaranteed to work in certain environments, and users are encouraged to install the new wheels that
+   come *with* the ``-cuXX`` suffix.
+
+   The argument ``--no-cache-dir`` is required for pip 23.1+. It forces pip to execute the
+   auto-detection logic.
+
+
+Citing cuQuantum
+================
+
+`H. Bayraktar et al., "cuQuantum SDK: A High-Performance Library for Accelerating Quantum Science," 2023 IEEE International Conference on Quantum Computing and Engineering (QCE), Bellevue, WA, USA, 2023, pp. 1050-1061, doi: 10.1109/QCE57702.2023.00119 <https://doi.org/10.1109/QCE57702.2023.00119>`_

cuquantum_python_jax-0.0.7/README.rst

@@ -0,0 +1,37 @@
+**********************
+cuQuantum Python JAX
+**********************
+
+cuQuantum Python JAX provides a JAX extension for cuQuantum Python. It exposes selected functionalities of
+cuQuantum SDK in a JAX-compatible way that enables JAX frameworks to directly interface with the exposed
+cuQuantum API. In the current release, cuQuantum JAX exposes JAX interfaces to the Operator Action API from
+the cuDensityMat library and the GF(2) sparse-dense matrix multiplication from the cuStabilizer library.
+
+Documentation
+=============
+
+For detailed guide, please refer to `cuQuantum Python JAX documentation <https://docs.nvidia.com/cuda/cuquantum/latest/python/jax.html>`_.
+
+Installation
+============
+
+.. code-block:: bash
+
+   pip install -v --no-cache-dir cuquantum-python-jax
+
+.. note::
+
+   This package is a meta package pointing to ``cuquantum-python-jax-cuXX``,
+   where XX is the CUDA major version (currently CUDA 12 & 13 are supported).
+   The meta package will attempt to infer and install the correct ``-cuXX`` wheel.
+   The auto-detection mechanism is not guaranteed to work in certain environments, and users are encouraged to install the new wheels that
+   come *with* the ``-cuXX`` suffix.
+
+   The argument ``--no-cache-dir`` is required for pip 23.1+. It forces pip to execute the
+   auto-detection logic.
+
+
+Citing cuQuantum
+================
+
+`H. Bayraktar et al., "cuQuantum SDK: A High-Performance Library for Accelerating Quantum Science," 2023 IEEE International Conference on Quantum Computing and Engineering (QCE), Bellevue, WA, USA, 2023, pp. 1050-1061, doi: 10.1109/QCE57702.2023.00119 <https://doi.org/10.1109/QCE57702.2023.00119>`_

cuquantum_python_jax-0.0.7/cuda_autodetect.py

@@ -0,0 +1,268 @@
+# Copyright (c) 2023, NVIDIA CORPORATION & AFFILIATES
+#
+# SPDX-License-Identifier: BSD-3-Clause
+#
+# =============================================================================
+#
+# Heavily influenced by https://github.com/cupy/cupy/blob/main/install/universal_pkg/setup.py.
+# See also the discussion and refs in https://github.com/NVIDIA/cuda-python/issues/16.
+# Below is the original copyright notice from cupy-wheel setup.py.
+#
+# Copyright (c) 2015-2023 Preferred Networks, Inc.
+#
+# SPDX-License-Identifier: MIT
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+import ctypes
+import importlib.metadata
+import os
+import sys
+from typing import Dict, List, Optional
+
+try:
+    from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
+except ImportError:
+    _bdist_wheel = None
+
+
+# This script is a general utility script, the variables below are to be set by
+# the caller
+PACKAGE_NAME = ''
+PACKAGE_SUPPORTED_CUDA_VER = []
+
+# ========================================================================
+
+PACKAGE_RESOLUTION = None
+CUDA_RESOLUTION = None
+
+
+class AutoDetectionFailed(Exception):
+    def __str__(self) -> str:
+        return f'''
+\n\n============================================================
+{super().__str__()}
+============================================================\n
+'''
+
+
+def _log(msg: str) -> None:
+    sys.stdout.write(f'[{PACKAGE_NAME}] {msg}\n')
+    sys.stdout.flush()
+
+
+def _get_version_from_library(
+        libnames: List[str],
+        funcname: str,
+        nvrtc: bool = False,
+) -> Optional[int]:
+    """Returns the library version from list of candidate libraries."""
+
+    for libname in libnames:
+        try:
+            _log(f'Looking for library: {libname}')
+            runtime_so = ctypes.CDLL(libname)
+            break
+        except Exception as e:
+            _log(f'Failed to open {libname}: {e}')
+    else:
+        _log('No more candidate library to find')
+        return None
+
+    func = getattr(runtime_so, funcname, None)
+    if func is None:
+        raise AutoDetectionFailed(
+            f'{libname}: {func} could not be found')
+    func.restype = ctypes.c_int
+
+    if nvrtc:
+        # nvrtcVersion
+        func.argtypes = [
+            ctypes.POINTER(ctypes.c_int),
+            ctypes.POINTER(ctypes.c_int),
+        ]
+        major = ctypes.c_int()
+        minor = ctypes.c_int()
+        retval = func(major, minor)
+        version = major.value * 1000 + minor.value * 10
+    else:
+        # cudaRuntimeGetVersion
+        func.argtypes = [
+            ctypes.POINTER(ctypes.c_int),
+        ]
+        version_ref = ctypes.c_int()
+        retval = func(version_ref)
+        version = version_ref.value
+
+    if retval != 0:  # NVRTC_SUCCESS or cudaSuccess
+        raise AutoDetectionFailed(
+            f'{libname}: {func} returned error: {retval}')
+    _log(f'Detected version: {version}')
+    return version
+
+
+def _get_cuda_version() -> Optional[int]:
+    """Returns the detected CUDA version or None."""
+
+    version = None
+
+    # First try NVRTC
+    libnames = [
+        'libnvrtc.so.13',
+        'libnvrtc.so.12',
+        'libnvrtc.so.11.2',
+        'libnvrtc.so.11.1',
+        'libnvrtc.so.11.0',
+    ]
+    _log(f'Trying to detect CUDA version from libraries: {libnames}')
+    try:
+        version = _get_version_from_library(libnames, 'nvrtcVersion', True)
+    except Exception as e:
+        _log(f"Error: {e}")  # log and move on
+    if version is not None:
+        return version
+
+    # Next try CUDART
+    libnames = [
+        'libcudart.so.13',
+        'libcudart.so.12',
+        'libcudart.so.11.0',  # side-effect: a CUDA context would be initialized
+    ]
+    _log(f'Trying to detect CUDA version from libraries: {libnames}')
+    try:
+        version = _get_version_from_library(libnames, 'cudaRuntimeGetVersion', False)
+    except Exception as e:
+        _log(f"Error: {e}")  # log and move on
+    if version is not None:
+        return version
+
+    _log("Autodetection failed")
+    return None
+
+
+def _find_installed_packages() -> List[str]:
+    """Returns the list of out packages installed in the environment."""
+
+    f = lambda x: ''.join([f"{PACKAGE_NAME}-cu", x])
+    found = []
+
+    for pkg in list(map(f, PACKAGE_SUPPORTED_CUDA_VER)):
+        try:
+            importlib.metadata.distribution(pkg)
+            found.append(pkg)
+        except importlib.metadata.PackageNotFoundError:
+            pass
+    return found
+
+
+def _cuda_version_to_package(ver: int) -> str:
+    # TODO: Don't hard-code 11/12, use PACKAGE_SUPPORTED_CUDA_VER instead?
+    if ver < 11000:
+        raise AutoDetectionFailed(
+            f'Your CUDA version ({ver}) is too old.')
+    elif ver < 12000:
+        # CUDA 11.x
+        raise RuntimeError(f'CUDA 11.x is not supported. Please install CUDA 12.x or later.')
+    elif ver < 13000:
+        # CUDA 12.x
+        suffix = '12'
+    elif ver < 14000:
+        # CUDA 13.x
+        suffix = '13'
+    else:
+        raise AutoDetectionFailed(
+            f'Your CUDA version ({ver}) is too new.')
+    return f'{PACKAGE_NAME}-cu{suffix}'
+
+
+# ========================================================================
+
+# "Public" API to the caller
+def infer_best_package(package_name: str,
+                       package_supported_cuda_ver: List[str] = ['12', '13']) -> str:
+    """Returns the appropriate wheel name for the environment."""
+
+    global PACKAGE_NAME, PACKAGE_SUPPORTED_CUDA_VER
+    PACKAGE_NAME = package_name
+    PACKAGE_SUPPORTED_CUDA_VER = sorted(package_supported_cuda_ver)
+
+    # Find the existing wheel installation
+    installed = _find_installed_packages()
+
+    # Detect CUDA version
+    version = _get_cuda_version()
+    if version is not None:
+        to_install = _cuda_version_to_package(version)
+    else:
+        # TODO: change this in the future
+        message = (
+            "See below for the error message and instruction.\n\n\n" +
+            "************************************************************************\n" +
+            "ERROR: Unable to detect NVIDIA CUDA Toolkit installation.\n" +
+            "ERROR: If CUDA Toolkit is not installed, please install it first.\n" + 
+            "ERROR: If CUDA Toolkit is installed but not detected, please explicitly specify the version and run\n" +
+            f"ERROR: `pip install {PACKAGE_NAME}-cuXX`, with XX being the major\n" +
+            "ERROR: version of your CUDA Toolkit installation.\n" +
+            "************************************************************************\n\n"
+        )
+        raise AutoDetectionFailed(message)
+
+    # Disallow -cu12 & -cu13 wheels from coexisting
+    if len(installed) > 1 or (len(installed) == 1 and installed[0] != to_install):
+        raise AutoDetectionFailed(
+            f'You already have the {PACKAGE_NAME} package(s) installed: \n'
+            f'  {installed}\n'
+            'while you attempt to install:\n'
+            f'  {to_install}\n'
+            'Please uninstall all of them first, then try reinstalling.')
+
+    global PACKAGE_RESOLUTION, CUDA_RESOLUTION
+    PACKAGE_RESOLUTION = to_install
+    CUDA_RESOLUTION = version
+    _log(f"Installing {to_install}...")
+    return to_install
+
+
+# "Public" API to the caller
+if _bdist_wheel is not None:
+
+    # Technically we need a way to force reinstalling the sdist and ignored the cached wheel.
+    # That said, I cannot reproduce the past-known behavior in my env. The sdist is always
+    # reinstalled, though it could be due to certain combination of pip/setuptools/wheel/etc,
+    # and it's still better to keep this WAR.
+
+    class bdist_wheel(_bdist_wheel):
+
+        # Adopted from https://discuss.python.org/t/wheel-caching-and-non-deterministic-builds/7687
+
+        def finalize_options(self):
+            super().finalize_options()
+
+            # Use "cuXX" as the build tag to force re-running sdist if the
+            # CUDA version in the user env has changed
+            if PACKAGE_RESOLUTION is None:
+                assert False, "something went wrong"
+            build_tag = PACKAGE_RESOLUTION.split("-")[-1]
+
+            # per PEP 427, build tag must start with a digit
+            self.build_number = f"0_{build_tag}"
+
+else:
+
+    bdist_wheel = None

cuquantum_python_jax-0.0.7/cuquantum_python_jax.egg-info/PKG-INFO

@@ -0,0 +1,79 @@
+Metadata-Version: 2.4
+Name: cuquantum-python-jax
+Version: 0.0.7
+Summary: NVIDIA cuQuantum Python JAX
+Home-page: https://developer.nvidia.com/cuquantum-sdk
+Author: NVIDIA Corporation
+Author-email: cuquantum-python@nvidia.com
+License: BSD-3-Clause
+Project-URL: Bug Tracker, https://github.com/NVIDIA/cuQuantum/issues
+Project-URL: User Forum, https://github.com/NVIDIA/cuQuantum/discussions
+Project-URL: Documentation, https://docs.nvidia.com/cuda/cuquantum/latest/python/jax.html
+Project-URL: Source Code, https://github.com/NVIDIA/cuQuantum
+Keywords: cuda,nvidia,state vector,tensor network,high-performance computing,quantum computing
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Topic :: Education
+Classifier: Topic :: Scientific/Engineering
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Environment :: GPU :: NVIDIA CUDA
+Classifier: Environment :: GPU :: NVIDIA CUDA :: 12
+Classifier: Environment :: GPU :: NVIDIA CUDA :: 13
+Requires-Python: >=3.11
+Description-Content-Type: text/x-rst
+License-File: LICENSE
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: requires-python
+Dynamic: summary
+
+**********************
+cuQuantum Python JAX
+**********************
+
+cuQuantum Python JAX provides a JAX extension for cuQuantum Python. It exposes selected functionalities of
+cuQuantum SDK in a JAX-compatible way that enables JAX frameworks to directly interface with the exposed
+cuQuantum API. In the current release, cuQuantum JAX exposes JAX interfaces to the Operator Action API from
+the cuDensityMat library and the GF(2) sparse-dense matrix multiplication from the cuStabilizer library.
+
+Documentation
+=============
+
+For detailed guide, please refer to `cuQuantum Python JAX documentation <https://docs.nvidia.com/cuda/cuquantum/latest/python/jax.html>`_.
+
+Installation
+============
+
+.. code-block:: bash
+
+   pip install -v --no-cache-dir cuquantum-python-jax
+
+.. note::
+
+   This package is a meta package pointing to ``cuquantum-python-jax-cuXX``,
+   where XX is the CUDA major version (currently CUDA 12 & 13 are supported).
+   The meta package will attempt to infer and install the correct ``-cuXX`` wheel.
+   The auto-detection mechanism is not guaranteed to work in certain environments, and users are encouraged to install the new wheels that
+   come *with* the ``-cuXX`` suffix.
+
+   The argument ``--no-cache-dir`` is required for pip 23.1+. It forces pip to execute the
+   auto-detection logic.
+
+
+Citing cuQuantum
+================
+
+`H. Bayraktar et al., "cuQuantum SDK: A High-Performance Library for Accelerating Quantum Science," 2023 IEEE International Conference on Quantum Computing and Engineering (QCE), Bellevue, WA, USA, 2023, pp. 1050-1061, doi: 10.1109/QCE57702.2023.00119 <https://doi.org/10.1109/QCE57702.2023.00119>`_

cuquantum_python_jax-0.0.7/cuquantum_python_jax.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+LICENSE
+README.rst
+cuda_autodetect.py
+setup.py
+cuquantum_python_jax.egg-info/PKG-INFO
+cuquantum_python_jax.egg-info/SOURCES.txt
+cuquantum_python_jax.egg-info/dependency_links.txt
+cuquantum_python_jax.egg-info/not-zip-safe
+cuquantum_python_jax.egg-info/top_level.txt

cuquantum_python_jax-0.0.7/cuquantum_python_jax.egg-info/top_level.txt

@@ -0,0 +1 @@
+cuda_autodetect

cuquantum_python_jax-0.0.7/setup.py

@@ -0,0 +1,99 @@
+# Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES.
+#
+# SPDX-License-Identifier: BSD-3-Clause
+
+import glob
+import os
+import shutil
+import site
+import subprocess
+import sys
+
+from setuptools import setup
+
+import cuda_autodetect
+from cuda_autodetect import infer_best_package, bdist_wheel
+
+
+# Update this for every release
+# Must pin to the cuquantum-python-jax-cuXX version
+package_ver = "0.0.7"
+package_name = "cuquantum-python-jax"
+
+
+# get project long description
+with open("README.rst") as f:
+    long_description = f.read()
+
+
+# This setup.py handles 2 cases:
+#   1. At the release time, we use it to generate sdist (which contains this script)
+#   2. At the install time, users install the sdist from PyPI
+# and the two cases have different requirements. We distinguish them by
+# setting CUQUANTUM_META_WHEEL_BUILD=1 for Case 1.
+if os.environ.get('CUQUANTUM_META_WHEEL_BUILD', '0') == '1':
+    # Case 1: generate sdist
+    install_requires = []
+    data_files = [('', ['cuda_autodetect.py',])]  # extra files to be copied into sdist
+    cmdclass = {}
+else:
+    # Case 2: install sdist
+    install_requires = [f"{infer_best_package(package_name)}=={package_ver}",]
+    if 12000 <= cuda_autodetect.CUDA_RESOLUTION < 13000:
+        # no ambiguity for CUDA 12, cuquantum-python-jax-cu12 would do the right thing
+        pass
+    elif 13000 <= cuda_autodetect.CUDA_RESOLUTION < 14000:
+        # no ambiguity for CUDA 13, cuquantum-python-jax-cu13 would do the right thing
+        pass
+    else:
+        raise ValueError(f"Unsupported CUDA version: {cuda_autodetect.CUDA_RESOLUTION}")
+    data_files = []
+    cmdclass = {'bdist_wheel': bdist_wheel} if bdist_wheel is not None else {}
+
+
+setup(
+    name=package_name,
+    version=package_ver,
+    description="NVIDIA cuQuantum Python JAX",
+    long_description=long_description,
+    long_description_content_type="text/x-rst",
+    url="https://developer.nvidia.com/cuquantum-sdk",
+    project_urls={
+        "Bug Tracker": "https://github.com/NVIDIA/cuQuantum/issues",
+        "User Forum": "https://github.com/NVIDIA/cuQuantum/discussions",
+        "Documentation": "https://docs.nvidia.com/cuda/cuquantum/latest/python/jax.html",
+        "Source Code": "https://github.com/NVIDIA/cuQuantum",
+    },
+    author="NVIDIA Corporation",
+    author_email="cuquantum-python@nvidia.com",
+    license="BSD-3-Clause",
+    license_files = ('LICENSE',),
+    keywords=["cuda", "nvidia", "state vector", "tensor network", "high-performance computing", "quantum computing"],
+    # Install files indicated by MANIFEST.in
+    # See https://github.com/pypa/sampleproject/issues/30#issuecomment-143947944
+    include_package_data=True,
+    zip_safe=False,
+    data_files=data_files,
+    setup_requires=[
+        "setuptools",
+        "wheel",
+    ],
+    install_requires=install_requires,
+    python_requires=">=3.11",
+    classifiers=[
+        "Development Status :: 5 - Production/Stable",
+        "Operating System :: POSIX :: Linux",
+        "Topic :: Education",
+        "Topic :: Scientific/Engineering",
+        "Programming Language :: Python :: 3 :: Only",
+        "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
+        "Programming Language :: Python :: 3.13",
+        "Programming Language :: Python :: 3.14",
+        "Programming Language :: Python :: Implementation :: CPython",
+        "Environment :: GPU :: NVIDIA CUDA",
+        "Environment :: GPU :: NVIDIA CUDA :: 12",
+        "Environment :: GPU :: NVIDIA CUDA :: 13",
+    ],
+    cmdclass=cmdclass,
+)

cuquantum_python_jax-0.0.4/MANIFEST.in

@@ -1,6 +0,0 @@
-include cuquantum/densitymat/jax/cppsrc/CMakeLists.txt
-include cuquantum/densitymat/jax/cppsrc/cudensitymat_jax.cpp
-include cuquantum/densitymat/jax/cppsrc/cudensitymat_jax.h
-include cuquantum/densitymat/jax/cppsrc/pybind.cpp
-include cuquantum/densitymat/jax/cppsrc/utils.h
-prune tests*