emu-base 2.0.1__py3-none-any.whl → 2.0.2__py3-none-any.whl

emu_base/__init__.py

@@ -16,4 +16,4 @@ __all__ = [
     "DEVICE_COUNT",
 ]
 
-__version__ = "2.0.1"
+__version__ = "2.0.2"

emu_base/math/double_krylov.py

@@ -0,0 +1,112 @@
+import torch
+from typing import Callable
+
+from emu_base.math.krylov_exp import DEFAULT_MAX_KRYLOV_DIM
+
+max_krylov_dim = DEFAULT_MAX_KRYLOV_DIM
+
+
+def double_krylov(
+    op: Callable,
+    state: torch.Tensor,
+    grad: torch.Tensor,
+    tolerance: float,
+) -> tuple[list[torch.Tensor], torch.Tensor, list[torch.Tensor]]:
+    """
+    Returns a Lanczos decomposition of the Fréchet derivative of the exponential
+    map U=exp(op) along the direction |state❭❬grad|.
+    The decomposition is represented by the tuple (Vs, dS, Vg) such that,
+        dU(op, |state❭❬grad|) = Vsᵗ @ dS @ Vg*
+
+    Args:
+        op (Callable): linear map to exponentiate, e.g. op(|ψ❭) = H|ψ❭.
+        state (torch.Tensor):
+        grad (torch.Tensor):
+        tolerance (float): tolerance of the returned derivative.
+
+    Returns:
+        Vstate (list): Lanczos basis of state
+        dS (torch.Tensor): matrix representing the derivative in the new basis
+        Vgrad (list): Lanczos basis of grad
+
+    Notes:
+        Fréchet derivative dU(op,|a❭❬b|) being defined as:
+            exp|op |a❭❬b|| = |exp(op) dU(op,|a❭❬b|)|
+               |0   op   |   |0       exp(op)      |
+
+        The function computes two Lanczos decomposition
+        up to the given tolerance
+            Va = Lanczos(|a❭,op(|a❭),op^2(|a❭),...)
+            Vb = Lanczos(|b❭,op(|b❭),op^2(|b❭),...)
+        such that,
+            op = Vaᵗ @ Ta @ Va*
+            op = Vbᵗ @ Tb @ Vb*
+
+        In the new basis Va, Vb
+            |op |a❭❬b|| -> |Ta  ab|0❭❬0||
+            |0   op   |    |0   Tb      |
+        where the top-right block only has one nonzero element.
+        Exponentiating such matrix and selecting the top-right block
+        gives us the desired matrix dS such that
+            dU(op, |a❭❬b|) = Vaᵗ @ dS @ Vb*
+    """
+    Vs, Ts = lanczos(op, state, tolerance)
+    Vg, Tg = lanczos(op, grad, tolerance)
+    size_s = len(Vs)
+    big_mat = torch.block_diag(Ts, Tg)
+    # Only one element in the top-right corner
+    big_mat[0, size_s] = state.norm() * grad.norm()
+    dS = torch.matrix_exp(big_mat)[:size_s, size_s:]
+    return Vs, dS, Vg
+
+
+def lanczos(
+    op: Callable,
+    v: torch.Tensor,
+    tolerance: float,
+) -> tuple[list[torch.Tensor], torch.Tensor]:
+    """
+    Copy of the code in krylov_exp to do Laczos iteration
+    To decide
+     1. refactor this
+     2. allow krylov results to store lanczos_vectors and T
+    """
+    converged = False
+    lanczos_vectors = [v / v.norm()]
+    T = torch.zeros(max_krylov_dim + 2, max_krylov_dim + 2, dtype=v.dtype)
+
+    for j in range(max_krylov_dim):
+        w = op(lanczos_vectors[-1])
+        n = w.norm()
+        for k in range(max(0, j - 1), j + 1):
+            overlap = torch.tensordot(lanczos_vectors[k].conj(), w, dims=w.dim())
+            T[k, j] = overlap
+            w -= overlap * lanczos_vectors[k]
+
+        n2 = w.norm()
+        T[j + 1, j] = n2
+
+        if n2 < tolerance:
+            converged = True
+            break
+
+        lanczos_vectors.append(w / n2)
+        # Compute exponential of extended T matrix
+        T[j + 2, j + 1] = 1
+        expd = torch.linalg.matrix_exp(T[: j + 3, : j + 3])
+
+        # Local truncation error estimation
+        err1 = abs(expd[j + 1, 0])
+        err2 = abs(expd[j + 2, 0] * n)
+
+        err = err1 if err1 < err2 else (err1 * err2 / (err1 - err2))
+        if err < tolerance:
+            converged = True
+            break
+
+    if not converged:
+        raise RecursionError(
+            "Lanczos iteration did not converge to precision in allotted number of steps."
+        )
+    size = len(lanczos_vectors)
+    return lanczos_vectors, T[:size, :size]

{emu_base-2.0.1.dist-info → emu_base-2.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-base
-Version: 2.0.1
+Version: 2.0.2
 Summary: Pasqal base classes for emulators
 Project-URL: Documentation, https://pasqal-io.github.io/emulators/
 Project-URL: Repository, https://github.com/pasqal-io/emulators
@@ -26,7 +26,7 @@ Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.10
 Requires-Dist: pulser-core==1.4.*
-Requires-Dist: torch==2.5.0
+Requires-Dist: torch==2.7.0
 Description-Content-Type: text/markdown
 
 <div align="center">

{emu_base-2.0.1.dist-info → emu_base-2.0.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-emu_base/__init__.py,sha256=ZKjTNCGsq85QyEaQgRUOXC9QcPg3W5Kjix-98JCejzs,493
+emu_base/__init__.py,sha256=dRd4TzTdTmz9IgVlisaBWBMjstVpVASVnmvjjSMbncU,493
 emu_base/aggregators.py,sha256=bB-rldoDAErxQMpL715K5lpiabGOpkCY0GyxW7mfHuc,5000
 emu_base/constants.py,sha256=41LYkKLUCz-oxPbd-j7nUDZuhIbUrnez6prT0uR0jcE,56
 emu_base/lindblad_operators.py,sha256=Nsl1YrWb8IDM9Z50ucy2Ed44p_IRETnlbr6qaqAgV50,1629
@@ -6,7 +6,8 @@ emu_base/pulser_adapter.py,sha256=dRD80z_dVXkCjDBLRIkmqNGg5M78VEKkQuk3H5JdZSM,11
 emu_base/utils.py,sha256=RM8O0qfPAJfcdqqAojwEEKV7I3ZfVDklnTisTGhUg5k,233
 emu_base/math/__init__.py,sha256=6BbIytYV5uC-e5jLMtIErkcUl_PvfSNnhmVFY9Il8uQ,97
 emu_base/math/brents_root_finding.py,sha256=AVx6L1Il6rpPJWrLJ7cn6oNmJyZOPRgEaaZaubC9lsU,3711
+emu_base/math/double_krylov.py,sha256=-DUZ5R3g7CUMQWSET2MUxXZKObXgLNanwAtS5nX8T68,3677
 emu_base/math/krylov_exp.py,sha256=UCFNeq-j2ukgBsOPC9_Jiv1aqpy88SrslDLiCxIGBwk,3840
-emu_base-2.0.1.dist-info/METADATA,sha256=Aj7pVr9uE3PWpLShQNgHEknUSznis0ZNG799P7UyRp0,3522
-emu_base-2.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_base-2.0.1.dist-info/RECORD,,
+emu_base-2.0.2.dist-info/METADATA,sha256=jqdSM-agy3W_gomgmN3dXuKw2NajCGgRtQPIDz1M6Us,3522
+emu_base-2.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_base-2.0.2.dist-info/RECORD,,