cuquantum-python-jax-cu12 0.0.5__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.6/NV.LICENSE

@@ -0,0 +1,121 @@
+NVIDIA SOFTWARE LICENSE
+
+(version August 18, 2021)
+
+This license is a legal agreement between you and NVIDIA Corporation ("NVIDIA") and governs your use of the NVIDIA
+cuQuantum software and materials provided hereunder (“SOFTWARE”).
+
+This license can be accepted only by an adult of legal age of majority in the country in which the SOFTWARE is used. If you
+are under the legal age of majority, you must ask your parent or legal guardian to consent to this license. By taking delivery
+of the SOFTWARE, you affirm that you have reached the legal age of majority, you accept the terms of this license, and you
+take legal and financial responsibility for the actions of your permitted users.
+
+You agree to use the SOFTWARE only for purposes that are permitted by (a) this license, and (b) any applicable law, regulation
+or generally accepted practices or guidelines in the relevant jurisdictions.
+
+1. LICENSE. Subject to the terms of this license, NVIDIA grants you a non-exclusive limited license to: (a) install and use the
+SOFTWARE, and (b) distribute the SOFTWARE subject to the distribution requirements described in this license. NVIDIA
+reserves all rights, title and interest in and to the SOFTWARE not expressly granted to you under this license.
+
+2. DISTRIBUTION REQUIREMENTS. These are the distribution requirements for you to exercise the distribution grant:
+a. The terms under which you distribute the SOFTWARE must be consistent with the terms of this license, including (without
+limitation) terms relating to the license grant and license restrictions and protection of NVIDIA’s intellectual property rights.
+b. You agree to notify NVIDIA in writing of any known or suspected distribution or use of the SOFTWARE not in compliance
+with the requirements of this license, and to enforce the terms of your agreements with respect to distributed SOFTWARE.
+
+3. LIMITATIONS. Your license to use the SOFTWARE is restricted as follows:
+a. You may not reverse engineer, decompile or disassemble, or remove copyright or other proprietary notices from any
+portion of the SOFTWARE or copies of the SOFTWARE.
+b. You may not modify or create derivative works of any portion of the SOFTWARE.
+c. You may not bypass, disable, or circumvent any technical measure, encryption, security, digital rights management or
+authentication mechanism in the SOFTWARE.
+d. You may not use the SOFTWARE in any manner that would cause it to become subject to an open source software license.
+As examples, licenses that require as a condition of use, modification, and/or distribution that the SOFTWARE be (i) disclosed
+or distributed in source code form; (ii) licensed for the purpose of making derivative works; or (iii) redistributable at no charge.
+e. Unless you have an agreement with NVIDIA for this purpose, you may not use the SOFTWARE with any system or
+application where the use or failure of the system or application can reasonably be expected to threaten or result in personal
+injury, death, or catastrophic loss. Examples include use in avionics, navigation, military, medical, life support or other life
+critical applications. NVIDIA does not design, test or manufacture the SOFTWARE for these critical uses and NVIDIA shall not
+be liable to you or any third party, in whole or in part, for any claims or damages arising from such uses.
+f. You agree to defend, indemnify and hold harmless NVIDIA and its affiliates, and their respective employees, contractors,
+agents, officers and directors, from and against any and all claims, damages, obligations, losses, liabilities, costs or debt, fines,
+restitutions and expenses (including but not limited to attorney’s fees and costs incident to establishing the right of
+indemnification) arising out of or related to use of the SOFTWARE outside of the scope of this Agreement, or not in compliance
+with its terms.
+
+4. PRE-RELEASE. SOFTWARE versions identified as alpha, beta, preview, early access or otherwise as pre-release may not be
+fully functional, may contain errors or design flaws, and may have reduced or different security, privacy, availability, and
+reliability standards relative to commercial versions of NVIDIA software and materials. You may use a pre-release SOFTWARE
+version at your own risk, understanding that these versions are not intended for use in production or business-critical
+systems.
+
+5. OWNERSHIP. The SOFTWARE and the related intellectual property rights therein are and will remain the sole and exclusive
+property of NVIDIA or its licensors. The SOFTWARE is copyrighted and protected by the laws of the United States and other
+countries, and international treaty provisions. NVIDIA may make changes to the SOFTWARE, at any time without notice, but
+is not obligated to support or update the SOFTWARE.
+
+6. COMPONENTS UNDER OTHER LICENSES. The SOFTWARE may include NVIDIA or third-party components with separate legal
+notices or terms as may be described in proprietary notices accompanying the SOFTWARE. If and to the extent there is a
+conflict between the terms in this license and the license terms associated with a component, the license terms associated
+with the components control only to the extent necessary to resolve the conflict.
+
+7. FEEDBACK. You may, but don’t have to, provide to NVIDIA any Feedback. “Feedback” means any suggestions, bug fixes,
+enhancements, modifications, feature requests or other feedback regarding the SOFTWARE. For any Feedback that you
+voluntarily provide, you hereby grant NVIDIA and its affiliates a perpetual, non-exclusive, worldwide, irrevocable license to
+use, reproduce, modify, license, sublicense (through multiple tiers of sublicensees), and distribute (through multiple tiers of
+distributors) the Feedback without the payment of any royalties or fees to you. NVIDIA will use Feedback at its choice.
+
+8. NO WARRANTIES. THE SOFTWARE IS PROVIDED "AS IS" WITHOUT ANY EXPRESS OR IMPLIED WARRANTY OF ANY KIND
+INCLUDING, BUT NOT LIMITED TO, WARRANTIES OF MERCHANTABILITY, NONINFRINGEMENT, OR FITNESS FOR A PARTICULAR
+PURPOSE. NVIDIA DOES NOT WARRANT THAT THE SOFTWARE WILL MEET YOUR REQUIREMENTS OR THAT THE OPERATION
+THEREOF WILL BE UNINTERRUPTED OR ERROR-FREE, OR THAT ALL ERRORS WILL BE CORRECTED.
+
+9. LIMITATIONSOF LIABILITY. TO THE MAXIMUM EXTENT PERMITTED BY LAW, NVIDIA AND ITS AFFILIATES SHALL NOT BE LIABLE
+FOR ANY SPECIAL, INCIDENTAL, PUNITIVE OR CONSEQUENTIAL DAMAGES, OR ANY LOST PROFITS, PROJECT DELAYS, LOSS OF
+USE, LOSS OF DATA OR LOSS OF GOODWILL, OR THE COSTS OF PROCURING SUBSTITUTE PRODUCTS, ARISING OUT OF OR IN
+CONNECTION WITH THIS LICENSE OR THE USE OR PERFORMANCE OF THE SOFTWARE, WHETHER SUCH LIABILITY ARISES FROM
+ANY CLAIM BASED UPON BREACH OF CONTRACT, BREACH OF WARRANTY, TORT (INCLUDING NEGLIGENCE), PRODUCT
+LIABILITY OR ANY OTHER CAUSE OF ACTION OR THEORY OF LIABILITY, EVEN IF NVIDIA HAS PREVIOUSLY BEEN ADVISED OF,
+OR COULD REASONABLY HAVE FORESEEN, THE POSSIBILITY OF SUCH DAMAGES. IN NO EVENT WILL NVIDIA’S AND ITS
+AFFILIATES TOTAL CUMULATIVE LIABILITY UNDER OR ARISING OUT OF THIS LICENSE EXCEED US$10.00. THE NATURE OF THE
+LIABILITY OR THE NUMBER OF CLAIMS OR SUITS SHALL NOT ENLARGE OR EXTEND THIS LIMIT.
+
+10. TERMINATION. Your rights under this license will terminate automatically without notice from NVIDIA if you fail to comply
+with any term and condition of this license or if you commence or participate in any legal proceeding against NVIDIA with
+respect to the SOFTWARE. NVIDIA may terminate this license with advance written notice to you if NVIDIA decides to no
+longer provide the SOFTWARE in a country or, in NVIDIA’s sole discretion, the continued use of it is no longer commercially
+viable. Upon any termination of this license, you agree to promptly discontinue use of the SOFTWARE and destroy all copies
+in your possession or control. Your prior distributions in accordance with this license are not affected by the termination of
+this license. All provisions of this license will survive termination, except for the license granted to you.
+
+11. APPLICABLE LAW. This license will be governed in all respects by the laws of the United States and of the State of Delaware
+as those laws are applied to contracts entered into and performed entirely within Delaware by Delaware residents, without
+regard to the conflicts of laws principles. The United Nations Convention on Contracts for the International Sale of Goods is
+specifically disclaimed. You agree to all terms of this Agreement in the English language. The state or federal courts residing
+in Santa Clara County, California shall have exclusive jurisdiction over any dispute or claim arising out of this license.
+Notwithstanding this, you agree that NVIDIA shall still be allowed to apply for injunctive remedies or an equivalent type of
+urgent legal relief in any jurisdiction.
+
+12. NO ASSIGNMENT. This license and your rights and obligations thereunder may not be assigned by you by any means or
+operation of law without NVIDIA’s permission. Any attempted assignment not approved by NVIDIA in writing shall be void
+and of no effect.
+
+13. EXPORT. The SOFTWARE is subject to United States export laws and regulations. You agree that you will not ship, transfer
+or export the SOFTWARE into any country, or use the SOFTWARE in any manner, prohibited by the United States Bureau of
+Industry and Security or economic sanctions regulations administered by the U.S. Department of Treasury’s Office of Foreign
+Assets Control (OFAC), or any applicable export laws, restrictions or regulations. These laws include restrictions on
+destinations, end users and end use. By accepting this license, you confirm that you are not a resident or citizen of any country
+currently embargoed by the U.S. and that you are not otherwise prohibited from receiving the SOFTWARE.
+
+14. GOVERNMENT USE. The SOFTWARE has been developed entirely at private expense and is “commercial items” consisting
+of “commercial computer software” and “commercial computer software documentation” provided with RESTRICTED
+RIGHTS. Use, duplication or disclosure by the U.S. Government or a U.S. Government subcontractor is subject to the
+restrictions in this license pursuant to DFARS 227.7202-3(a) or as set forth in subparagraphs (b)(1) and (2) of the Commercial
+Computer Software - Restricted Rights clause at FAR 52.227-19, as applicable. Contractor/manufacturer is NVIDIA, 2788 San
+Tomas Expressway, Santa Clara, CA 95051.
+
+15. ENTIRE AGREEMENT. This license is the final, complete and exclusive agreement between the parties relating to the
+subject matter of this license and supersedes all prior or contemporaneous understandings and agreements relating to this
+subject matter, whether oral or written. If any court of competent jurisdiction determines that any provision of this license is
+illegal, invalid or unenforceable, the remaining provisions will remain in full force and effect. This license may only be modified
+in a writing signed by an authorized representative of each party.

{cuquantum_python_jax_cu12-0.0.5/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
+Version: 0.0.6
 Summary: NVIDIA cuQuantum Python JAX
 Author-email: NVIDIA Corporation <cuquantum-python@nvidia.com>
 License-Expression: BSD-3-Clause
@@ -13,20 +13,21 @@ 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
 Requires-Python: >=3.11.0
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: pybind11
-Requires-Dist: cuquantum-python-cu12~=26.1.0
-Requires-Dist: jax[cuda12-local]<0.7,>=0.5
+License-File: NV.LICENSE
+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
 
@@ -36,17 +37,17 @@ Please visit the [NVIDIA cuQuantum Python documentation](https://docs.nvidia.com
 
 ### Requirements
 
-The install-time dependencies of the cuQuantum Python JAX package include:
+The build-time dependencies of the cuQuantum Python JAX package include:
 
-* cuquantum-python-cu12~=26.1.0 for CUDA 12 or cuquantum-python-cu13~=26.1.0 for CUDA 13
+* 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
 * setuptools>=77.0.3
 
 Note:
-1. cuQuantum Python JAX is only supported with CUDA 12 and CUDA 13.
-2. cuQuantum Python JAX installation does not support build isolation. The user needs to pass in `--no-build-isolation` to `pip` when installing cuQuantum Python JAX.
-3. 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.
+- cuQuantum Python JAX wheels are CUDA-versioned: `cuquantum-python-jax-cu12` for CUDA 12 and `cuquantum-python-jax-cu13` for CUDA 13.
 
 #### Installation using `jax[cudaXX-local]`
 
@@ -68,26 +69,25 @@ pip install nvidia-cuda-cupti-cu12
 After installing cuDNN and cuPTI, the user may install cuQuantum Python JAX with `pip` using either:
 
 ```
-pip install --no-build-isolation cuquantum-python-jax-cu12   # for CUDA 12
-pip install --no-build-isolation cuquantum-python-jax-cu13   # for CUDA 13
+pip install cuquantum-python-jax-cu12   # for CUDA 12
+pip install cuquantum-python-jax-cu13   # for CUDA 13
 ```
 
 or one of
 
 ```
-pip install --no-build-isolation cuquantum-python-cu12[jax]
-pip install --no-build-isolation cuquantum-python-cu13[jax]
+pip install cuquantum-python-cu12[jax]
+pip install cuquantum-python-cu13[jax]
 ```
 
 where the CUDA version is explicitly specified on cuquantum-python.
 
 Note:
-1. If cuDNN and cuPTI are installed with `pip`, the user does not need to specify library folders in `LD_LIBRARY_PATH`.
-2. When the latter command `pip install --no-build-isolation cuquantum-python-cu12[jax]`/`pip install --no-build-isolation cuquantum-python-cu13[jax]` is used, `--no-build-isolation` applies to both cuquantum-python and cuquantum-python-jax. The user needs to ensure cuquantum-python's build dependencies are installed before the installation.
+- If cuDNN and cuPTI are installed with `pip`, the user does not need to specify library folders in `LD_LIBRARY_PATH`.
 
 #### Installing from source
 
-To install cuQuantum Python JAX from source, first compile cuQuantum Python from source using the [instructions on GitHub](https://github.com/NVIDIA/cuQuantum/blob/main/python/README.md). Once complete, navigate to `python/extensions`, then:
+To install cuQuantum Python JAX from source, first compile cuQuantum Python from source using the [instructions on GitHub](https://github.com/NVIDIA/cuQuantum/blob/main/python/README.md). Once complete, navigate to `python/extensions`, run `./configure.sh` to generate a CUDA version-specific `pyproject.toml` from the template, and then:
 
 ```
 pip install .
@@ -102,9 +102,9 @@ 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+
-* cuquantum-python-cu12~=26.1.0 for CUDA 12 or cuquantum-python-cu13~=26.1.0 for CUDA 13
+* 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
-* pybind11
 
 ## Developer Notes
 

{cuquantum_python_jax_cu12-0.0.5 → 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
 
@@ -10,17 +10,17 @@ Please visit the [NVIDIA cuQuantum Python documentation](https://docs.nvidia.com
 
 ### Requirements
 
-The install-time dependencies of the cuQuantum Python JAX package include:
+The build-time dependencies of the cuQuantum Python JAX package include:
 
-* cuquantum-python-cu12~=26.1.0 for CUDA 12 or cuquantum-python-cu13~=26.1.0 for CUDA 13
+* 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
 * setuptools>=77.0.3
 
 Note:
-1. cuQuantum Python JAX is only supported with CUDA 12 and CUDA 13.
-2. cuQuantum Python JAX installation does not support build isolation. The user needs to pass in `--no-build-isolation` to `pip` when installing cuQuantum Python JAX.
-3. 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.
+- cuQuantum Python JAX wheels are CUDA-versioned: `cuquantum-python-jax-cu12` for CUDA 12 and `cuquantum-python-jax-cu13` for CUDA 13.
 
 #### Installation using `jax[cudaXX-local]`
 
@@ -42,26 +42,25 @@ pip install nvidia-cuda-cupti-cu12
 After installing cuDNN and cuPTI, the user may install cuQuantum Python JAX with `pip` using either:
 
 ```
-pip install --no-build-isolation cuquantum-python-jax-cu12   # for CUDA 12
-pip install --no-build-isolation cuquantum-python-jax-cu13   # for CUDA 13
+pip install cuquantum-python-jax-cu12   # for CUDA 12
+pip install cuquantum-python-jax-cu13   # for CUDA 13
 ```
 
 or one of
 
 ```
-pip install --no-build-isolation cuquantum-python-cu12[jax]
-pip install --no-build-isolation cuquantum-python-cu13[jax]
+pip install cuquantum-python-cu12[jax]
+pip install cuquantum-python-cu13[jax]
 ```
 
 where the CUDA version is explicitly specified on cuquantum-python.
 
 Note:
-1. If cuDNN and cuPTI are installed with `pip`, the user does not need to specify library folders in `LD_LIBRARY_PATH`.
-2. When the latter command `pip install --no-build-isolation cuquantum-python-cu12[jax]`/`pip install --no-build-isolation cuquantum-python-cu13[jax]` is used, `--no-build-isolation` applies to both cuquantum-python and cuquantum-python-jax. The user needs to ensure cuquantum-python's build dependencies are installed before the installation.
+- If cuDNN and cuPTI are installed with `pip`, the user does not need to specify library folders in `LD_LIBRARY_PATH`.
 
 #### Installing from source
 
-To install cuQuantum Python JAX from source, first compile cuQuantum Python from source using the [instructions on GitHub](https://github.com/NVIDIA/cuQuantum/blob/main/python/README.md). Once complete, navigate to `python/extensions`, then:
+To install cuQuantum Python JAX from source, first compile cuQuantum Python from source using the [instructions on GitHub](https://github.com/NVIDIA/cuQuantum/blob/main/python/README.md). Once complete, navigate to `python/extensions`, run `./configure.sh` to generate a CUDA version-specific `pyproject.toml` from the template, and then:
 
 ```
 pip install .
@@ -76,9 +75,9 @@ 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+
-* cuquantum-python-cu12~=26.1.0 for CUDA 12 or cuquantum-python-cu13~=26.1.0 for CUDA 13
+* 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
-* pybind11
 
 ## Developer Notes
 

{cuquantum_python_jax_cu12-0.0.5 → 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 → 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)