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

{cuquantum_python_jax-0.0.3/cuquantum_python_jax.egg-info → cuquantum_python_jax-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cuquantum-python-jax
-Version: 0.0.3
+Version: 0.0.4
 Summary: NVIDIA cuQuantum Python JAX
 Home-page: https://developer.nvidia.com/cuquantum-sdk
 Author: NVIDIA Corporation
@@ -16,13 +16,13 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Environment :: GPU :: NVIDIA CUDA
-Classifier: Environment :: GPU :: NVIDIA CUDA :: 13
+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-cu13~=25.11
-Requires-Dist: jax[cuda13-local]<0.9,>=0.8
+Requires-Dist: cuquantum-python-cu12~=26.1.0
+Requires-Dist: jax[cuda12-local]<0.7,>=0.5
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -47,14 +47,16 @@ Please visit the [NVIDIA cuQuantum Python documentation](https://docs.nvidia.com
 
 ### Requirements
 
-The install-time dependencies of the cuQuantum Python package include:
+The install-time dependencies of the cuQuantum Python JAX package include:
 
-* cuquantum-python-cu12~=25.11 for CUDA 12 or cuquantum-python-cu13~=25.11 for CUDA 13
+* cuquantum-python-cu12~=26.1.0 for CUDA 12 or cuquantum-python-cu13~=26.1.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
 * setuptools>=77.0.3
 
-Note: cuQuantum Python JAX is only supported with CUDA 12 and CUDA 13.
+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.
 
 #### Installation using `jax[cudaXX-local]`
 
@@ -73,20 +75,24 @@ pip install nvidia-cudnn-cu12
 pip install nvidia-cuda-cupti-cu12
 ```
 
-After installing cuDNN and cuPTI, the user may install `cuquantum-python-jax` using `pip` using either:
+After installing cuDNN and cuPTI, the user may install `cuquantum-python-jax` with `pip` using either:
 
 ```
-pip install cuquantum-python-jax
+pip install --no-build-isolation cuquantum-python-jax
 ```
 
-or
+in which case the CUDA version will be detected, or one of
 
 ```
-pip install cuquantum-python-cu12[jax]
-pip install cuquantum-python-cu13[jax]
+pip install --no-build-isolation cuquantum-python-cu12[jax]
+pip install --no-build-isolation cuquantum-python-cu13[jax]
 ```
 
-Note: if cuDNN and cuPTI are installed with `pip`, the user does not need to specify library folders in `LD_LIBRARY_PATH`.
+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.
 
 #### Installing from source
 
@@ -106,8 +112,8 @@ Where `CUDENSITYMAT_ROOT` is the path to the libraries parent directory. For exa
 Runtime dependencies of the cuQuantum Python package include:
 
 * An NVIDIA GPU with compute capability 7.5+
-* cuquantum-python-cu12~=25.11 for CUDA 12 or cuquantum-python-cu13~=25.11 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-cu12~=26.1.0 for CUDA 12 or cuquantum-python-cu13~=26.1.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-0.0.3 → cuquantum_python_jax-0.0.4}/README.md

@@ -10,14 +10,16 @@ Please visit the [NVIDIA cuQuantum Python documentation](https://docs.nvidia.com
 
 ### Requirements
 
-The install-time dependencies of the cuQuantum Python package include:
+The install-time dependencies of the cuQuantum Python JAX package include:
 
-* cuquantum-python-cu12~=25.11 for CUDA 12 or cuquantum-python-cu13~=25.11 for CUDA 13
+* cuquantum-python-cu12~=26.1.0 for CUDA 12 or cuquantum-python-cu13~=26.1.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
 * setuptools>=77.0.3
 
-Note: cuQuantum Python JAX is only supported with CUDA 12 and CUDA 13.
+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.
 
 #### Installation using `jax[cudaXX-local]`
 
@@ -36,20 +38,24 @@ pip install nvidia-cudnn-cu12
 pip install nvidia-cuda-cupti-cu12
 ```
 
-After installing cuDNN and cuPTI, the user may install `cuquantum-python-jax` using `pip` using either:
+After installing cuDNN and cuPTI, the user may install `cuquantum-python-jax` with `pip` using either:
 
 ```
-pip install cuquantum-python-jax
+pip install --no-build-isolation cuquantum-python-jax
 ```
 
-or
+in which case the CUDA version will be detected, or one of
 
 ```
-pip install cuquantum-python-cu12[jax]
-pip install cuquantum-python-cu13[jax]
+pip install --no-build-isolation cuquantum-python-cu12[jax]
+pip install --no-build-isolation cuquantum-python-cu13[jax]
 ```
 
-Note: if cuDNN and cuPTI are installed with `pip`, the user does not need to specify library folders in `LD_LIBRARY_PATH`.
+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.
 
 #### Installing from source
 
@@ -69,8 +75,8 @@ Where `CUDENSITYMAT_ROOT` is the path to the libraries parent directory. For exa
 Runtime dependencies of the cuQuantum Python package include:
 
 * An NVIDIA GPU with compute capability 7.5+
-* cuquantum-python-cu12~=25.11 for CUDA 12 or cuquantum-python-cu13~=25.11 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-cu12~=26.1.0 for CUDA 12 or cuquantum-python-cu13~=26.1.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-0.0.3 → cuquantum_python_jax-0.0.4}/cuquantum/densitymat/jax/__init__.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES
+# Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES
 #
 # SPDX-License-Identifier: BSD-3-Clause
 
@@ -10,7 +10,7 @@ if not jax.config.jax_enable_x64:
     raise RuntimeError(f"jax_enable_x64 must be set to True to use cuQuantum Python JAX")
 
 from .operator_action import operator_action
-from .pysrc.operators import (
+from .pysrc import (
     ElementaryOperator,
     MatrixOperator,
     OperatorTerm,