cuquantum-python-jax-cu13 0.0.5.post0__tar.gz → 0.0.7__tar.gz

cuquantum_python_jax_cu13-0.0.7/MANIFEST.in

@@ -0,0 +1,44 @@
+#
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: BSD-3-Clause
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+# 1. Redistributions of source code must retain the above copyright notice, this
+# list of conditions and the following disclaimer.
+#
+# 2. Redistributions in binary form must reproduce the above copyright notice,
+# this list of conditions and the following disclaimer in the documentation
+# and/or other materials provided with the distribution.
+#
+# 3. Neither the name of the copyright holder nor the names of its
+# contributors may be used to endorse or promote products derived from
+# this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+
+include pyproject.toml.template
+include configure.sh
+include cuquantum/densitymat/jax/cppsrc/CMakeLists.txt
+include cuquantum/densitymat/jax/cppsrc/cudensitymat.h
+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
+include cuquantum/stabilizer/jax/cppsrc/CMakeLists.txt
+include cuquantum/stabilizer/jax/cppsrc/custabilizer.h
+include cuquantum/stabilizer/jax/cppsrc/custabilizer_jax.cpp
+include cuquantum/stabilizer/jax/cppsrc/custabilizer_jax.h
+include cuquantum/stabilizer/jax/cppsrc/pybind.cpp
+prune tests*

{cuquantum_python_jax_cu13-0.0.5.post0/cuquantum_python_jax_cu13.egg-info → cuquantum_python_jax_cu13-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cuquantum-python-jax-cu13
-Version: 0.0.5.post0
+Version: 0.0.7
 Summary: NVIDIA cuQuantum Python JAX
 Author-email: NVIDIA Corporation <cuquantum-python@nvidia.com>
 License-Expression: BSD-3-Clause
@@ -13,6 +13,7 @@ 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 :: 13
@@ -20,13 +21,13 @@ Requires-Python: >=3.11.0
 Description-Content-Type: text/markdown
 License-File: LICENSE
 License-File: NV.LICENSE
-Requires-Dist: cuquantum-python-cu13~=26.3.0
-Requires-Dist: jax[cuda13-local]<0.9,>=0.8
+Requires-Dist: cuquantum-python-cu13~=26.9.0
+Requires-Dist: jax[cuda13-local]>=0.8
 Dynamic: license-file
 
 # cuQuantum Python JAX
 
-cuQuantum Python JAX provides a JAX extension for cuQuantum Python. It exposes selected functionality 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 a JAX interface to the Operator Action API from the cuDensityMat library.
+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
 
@@ -38,14 +39,17 @@ Please visit the [NVIDIA cuQuantum Python documentation](https://docs.nvidia.com
 
 The build-time dependencies of the cuQuantum Python JAX package include:
 
-* jax[cuda12-local]>=0.5,<0.7 for CUDA 12 or jax[cuda13-local]>=0.8,<0.9 for CUDA 13
+* Python 3.11+
+* CUDA Toolkit 12.x or 13.x
+* cmake>=3.22
+* jax[cuda12-local]>=0.8 for CUDA 12 or jax[cuda13-local]>=0.8 for CUDA 13
 * pybind11
 * wheel
 * setuptools>=77.0.3
 
 Note:
-- cuQuantum Python JAX is only supported with CUDA 12 and CUDA 13.
-- cuQuantum Python JAX wheels are CUDA-versioned: `cuquantum-python-jax-cu12` for CUDA 12 and `cuquantum-python-jax-cu13` for CUDA 13.
+- cuQuantum Python JAX is only supported with CUDA 12 and CUDA 13 and the distributions are CUDA-versioned: `cuquantum-python-jax-cu12` for CUDA 12 and `cuquantum-python-jax-cu13` for CUDA 13.
+- **Build isolation note:** When installing with build isolation (`pip`'s default mode), the JAX version used at build time is determined by the build dependency specification in `pyproject.toml` and may differ from the JAX version pre-installed in the environment. Mismatched build-time and runtime JAX versions can lead to errors when importing cuQuantum's JAX extension. We recommend passing `--no-build-isolation` to `pip install` and pre-installing all build dependencies manually.
 
 #### Installation using `jax[cudaXX-local]`
 
@@ -99,9 +103,11 @@ The CUDA version is detected automatically from `$CUDA_PATH` and the wheel will
 
 Runtime dependencies of the cuQuantum Python JAX package include:
 
+* Python 3.11+
 * An NVIDIA GPU with compute capability 7.5+
-* cuquantum-python-cu12~=26.3.0 for CUDA 12 or cuquantum-python-cu13~=26.3.0 for CUDA 13
-* jax[cuda12-local]>=0.5,<0.7 for CUDA 12 or jax[cuda13-local]>=0.8,<0.9 for CUDA 13
+* CUDA Toolkit 12.x or 13.x
+* cuquantum-python-cu12~=26.9.0 for CUDA 12 or cuquantum-python-cu13~=26.9.0 for CUDA 13
+* jax[cuda12-local]>=0.8 for CUDA 12 or jax[cuda13-local]>=0.8 for CUDA 13
 
 ## Developer Notes
 

{cuquantum_python_jax_cu13-0.0.5.post0 → cuquantum_python_jax_cu13-0.0.7}/README.md

@@ -1,6 +1,6 @@
 # cuQuantum Python JAX
 
-cuQuantum Python JAX provides a JAX extension for cuQuantum Python. It exposes selected functionality 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 a JAX interface to the Operator Action API from the cuDensityMat library.
+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
 
@@ -12,14 +12,17 @@ Please visit the [NVIDIA cuQuantum Python documentation](https://docs.nvidia.com
 
 The build-time dependencies of the cuQuantum Python JAX package include:
 
-* jax[cuda12-local]>=0.5,<0.7 for CUDA 12 or jax[cuda13-local]>=0.8,<0.9 for CUDA 13
+* Python 3.11+
+* CUDA Toolkit 12.x or 13.x
+* cmake>=3.22
+* jax[cuda12-local]>=0.8 for CUDA 12 or jax[cuda13-local]>=0.8 for CUDA 13
 * pybind11
 * wheel
 * setuptools>=77.0.3
 
 Note:
-- cuQuantum Python JAX is only supported with CUDA 12 and CUDA 13.
-- cuQuantum Python JAX wheels are CUDA-versioned: `cuquantum-python-jax-cu12` for CUDA 12 and `cuquantum-python-jax-cu13` for CUDA 13.
+- cuQuantum Python JAX is only supported with CUDA 12 and CUDA 13 and the distributions are CUDA-versioned: `cuquantum-python-jax-cu12` for CUDA 12 and `cuquantum-python-jax-cu13` for CUDA 13.
+- **Build isolation note:** When installing with build isolation (`pip`'s default mode), the JAX version used at build time is determined by the build dependency specification in `pyproject.toml` and may differ from the JAX version pre-installed in the environment. Mismatched build-time and runtime JAX versions can lead to errors when importing cuQuantum's JAX extension. We recommend passing `--no-build-isolation` to `pip install` and pre-installing all build dependencies manually.
 
 #### Installation using `jax[cudaXX-local]`
 
@@ -73,9 +76,11 @@ The CUDA version is detected automatically from `$CUDA_PATH` and the wheel will
 
 Runtime dependencies of the cuQuantum Python JAX package include:
 
+* Python 3.11+
 * An NVIDIA GPU with compute capability 7.5+
-* cuquantum-python-cu12~=26.3.0 for CUDA 12 or cuquantum-python-cu13~=26.3.0 for CUDA 13
-* jax[cuda12-local]>=0.5,<0.7 for CUDA 12 or jax[cuda13-local]>=0.8,<0.9 for CUDA 13
+* CUDA Toolkit 12.x or 13.x
+* cuquantum-python-cu12~=26.9.0 for CUDA 12 or cuquantum-python-cu13~=26.9.0 for CUDA 13
+* jax[cuda12-local]>=0.8 for CUDA 12 or jax[cuda13-local]>=0.8 for CUDA 13
 
 ## Developer Notes
 

{cuquantum_python_jax_cu13-0.0.5.post0 → cuquantum_python_jax_cu13-0.0.7}/configure.sh

@@ -73,11 +73,11 @@ echo "Detected CUDA_VERSION=${CUDA_VERSION_RAW} (major=${CUDA_MAJOR}) from ${CUD
 # -------------------------------------------------------------------
 case "${CUDA_MAJOR}" in
     12)
-        JAX_VERSION_SPEC=">=0.5,<0.7"
+        JAX_VERSION_SPEC=">=0.8"
         CUDA_CLASSIFIER="Environment :: GPU :: NVIDIA CUDA :: 12"
         ;;
     13)
-        JAX_VERSION_SPEC=">=0.8,<0.9"
+        JAX_VERSION_SPEC=">=0.8"
         CUDA_CLASSIFIER="Environment :: GPU :: NVIDIA CUDA :: 13"
         ;;
     *)

cuquantum_python_jax_cu13-0.0.7/cuquantum/densitymat/jax/__init__.py

@@ -0,0 +1,20 @@
+# Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES
+#
+# SPDX-License-Identifier: BSD-3-Clause
+
+try:
+    from ._build_info import check_jax_abi as _check_jax_abi
+except ImportError:
+    pass
+else:
+    _check_jax_abi()
+    del _check_jax_abi
+
+from .operator_action import operator_action
+from .pysrc import (
+    ElementaryOperator,
+    MatrixOperator,
+    OperatorTerm,
+    Operator,
+    SimplifierConfig
+)

cuquantum_python_jax_cu13-0.0.7/cuquantum/densitymat/jax/_build_info.py

@@ -0,0 +1,19 @@
+# Auto-generated by setup.py at build time; do not edit.
+JAX_BUILD_VERSION = ""
+PROJECT_NAME = "cuquantum-python-jax-cu13"
+
+def check_jax_abi() -> None:
+    """Warn whenever the runtime jax version differs from the
+    build-time jax version. JAX uses effort-based versioning, so
+    any difference -- patch bumps included -- may shift the FFI
+    capsule layout these bindings were compiled against."""
+    if not JAX_BUILD_VERSION:
+        return
+    import jax, warnings
+    if jax.__version__ != JAX_BUILD_VERSION:
+        warnings.warn(
+            f"JAX version mismatch: {PROJECT_NAME} was built using "
+            f"JAX {JAX_BUILD_VERSION}, current runtime is JAX "
+            f"{jax.__version__}. Rebuild against the runtime jax if "
+            f"you see issues.",
+            RuntimeWarning, stacklevel=3)