cuquantum-python-jax-cu12 0.0.5.post0__tar.gz → 0.0.6__tar.gz

cuquantum_python_jax_cu12-0.0.6/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_cu12-0.0.5.post0/cuquantum_python_jax_cu12.egg-info → cuquantum_python_jax_cu12-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cuquantum-python-jax-cu12
-Version: 0.0.5.post0
+Version: 0.0.6
 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 :: 12
@@ -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-cu12~=26.3.0
-Requires-Dist: jax[cuda12-local]<0.7,>=0.5
+Requires-Dist: cuquantum-python-cu12~=26.6.0
+Requires-Dist: jax[cuda12-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 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 and the GF(2) sparse-dense matrix multiply from the cuStabilizer library.
 
 ## Documentation
 
@@ -38,6 +39,7 @@ Please visit the [NVIDIA cuQuantum Python documentation](https://docs.nvidia.com
 
 The build-time dependencies of the cuQuantum Python JAX package include:
 
+* CUDA Toolkit 12.x or 13.x
 * jax[cuda12-local]>=0.5,<0.7 for CUDA 12 or jax[cuda13-local]>=0.8,<0.9 for CUDA 13
 * pybind11
 * wheel
@@ -100,6 +102,7 @@ The CUDA version is detected automatically from `$CUDA_PATH` and the wheel will
 Runtime dependencies of the cuQuantum Python JAX package include:
 
 * An NVIDIA GPU with compute capability 7.5+
+* CUDA Toolkit 12.x or 13.x
 * 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
 

{cuquantum_python_jax_cu12-0.0.5.post0 → cuquantum_python_jax_cu12-0.0.6}/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 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 and the GF(2) sparse-dense matrix multiply from the cuStabilizer library.
 
 ## Documentation
 
@@ -12,6 +12,7 @@ Please visit the [NVIDIA cuQuantum Python documentation](https://docs.nvidia.com
 
 The build-time dependencies of the cuQuantum Python JAX package include:
 
+* CUDA Toolkit 12.x or 13.x
 * jax[cuda12-local]>=0.5,<0.7 for CUDA 12 or jax[cuda13-local]>=0.8,<0.9 for CUDA 13
 * pybind11
 * wheel
@@ -74,6 +75,7 @@ The CUDA version is detected automatically from `$CUDA_PATH` and the wheel will
 Runtime dependencies of the cuQuantum Python JAX package include:
 
 * An NVIDIA GPU with compute capability 7.5+
+* CUDA Toolkit 12.x or 13.x
 * 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
 

{cuquantum_python_jax_cu12-0.0.5.post0 → cuquantum_python_jax_cu12-0.0.6}/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_cu12-0.0.5.post0 → cuquantum_python_jax_cu12-0.0.6}/cuquantum/densitymat/jax/__init__.py

@@ -6,8 +6,17 @@ from cuquantum.bindings._internal import cudensitymat as _cudm
 _cudm._inspect_function_pointers()  # for loading libcudensitymat.so
 
 import jax
+
+try:
+    from ._build_info import check_jax_abi as _check_jax_abi
+except ImportError:
+    pass
+else:
+    _check_jax_abi()
+    del _check_jax_abi
+
 if not jax.config.jax_enable_x64:
-    raise RuntimeError(f"jax_enable_x64 must be set to True to use cuQuantum Python JAX")
+    raise RuntimeError("jax_enable_x64 must be set to True to use cuQuantum Python JAX")
 
 from .operator_action import operator_action
 from .pysrc import (

cuquantum_python_jax_cu12-0.0.6/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-cu12"
+
+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)