ket-lang 0.10.2.dev0__tar.gz → 0.10.2.dev1__tar.gz

{ket_lang-0.10.2.dev0/src/ket_lang.egg-info → ket_lang-0.10.2.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ket-lang
-Version: 0.10.2.dev0
+Version: 0.10.2.dev1
 Summary: Ket quantum programming language interpreter and library
 Home-page: https://quantumket.org
 Author: Evandro Chagas Ribeiro da Rosa

{ket_lang-0.10.2.dev0 → ket_lang-0.10.2.dev1}/src/ket/__init__.py

@@ -83,7 +83,7 @@ from .fermion import *
 from .fermion import __all__ as all_fermion
 from . import qulib
 
-__version__ = "0.10.2.dev0"
+__version__ = "0.10.2.dev1"
 
 __all__ = (
     all_base

{ket_lang-0.10.2.dev0 → ket_lang-0.10.2.dev1}/src/ket/mps.py

@@ -161,6 +161,8 @@ class MPS(BatchExecution):
             shapes and disables JAX's ``jit`` compilation, which may result in slower evaluations.
             If ``None``, the network is built in 2D and contracted fully at the end,
             which is JIT-compatible but may scale poorly with depth. Defaults to ``None``.
+        double_precision: Whether to enable 64-bit floating point precision (double precision) in
+            JAX. Defaults to ``False``. Set to ``True`` for higher accuracy in deep circuits.
     """
 
     def __init__(
@@ -168,13 +170,14 @@ class MPS(BatchExecution):
         num_qubits: int,
         backend: Literal["cpu", "gpu"] = "cpu",
         max_bond: int = None,
+        double_precision: bool = False,
     ):
         super().__init__()
         self.num_qubits = num_qubits
         self.max_bond = max_bond
         if backend not in ["cpu", "gpu"]:
             raise ValueError("backend must be 'cpu' or 'gpu'")
-        jax.config.update("jax_enable_x64", True)
+        jax.config.update("jax_enable_x64", double_precision)
         jax.config.update("jax_platform_name", backend)
 
         self.last_grad = None

{ket_lang-0.10.2.dev0 → ket_lang-0.10.2.dev1/src/ket_lang.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ket-lang
-Version: 0.10.2.dev0
+Version: 0.10.2.dev1
 Summary: Ket quantum programming language interpreter and library
 Home-page: https://quantumket.org
 Author: Evandro Chagas Ribeiro da Rosa