emu-base 2.0.3__py3-none-any.whl → 2.1.0__py3-none-any.whl

emu_base/__init__.py

@@ -16,4 +16,4 @@ __all__ = [
     "DEVICE_COUNT",
 ]
 
-__version__ = "2.0.3"
+__version__ = "2.1.0"

emu_base/{lindblad_operators.py → jump_lindblad_operators.py}

@@ -45,3 +45,16 @@ def get_lindblad_operators(
         ]
 
     raise ValueError(f"Unknown noise type: {noise_type}")
+
+
+def compute_noise_from_lindbladians(lindbladians: list[torch.Tensor]) -> torch.Tensor:
+    """
+    Compute the single-qubit Hamiltonian noise term -0.5i∑L†L from all the given lindbladians.
+    """
+    assert all(
+        lindbladian.shape == (2, 2) for lindbladian in lindbladians
+    ), "Only single-qubit lindblad operators are supported"
+
+    zero = torch.zeros(2, 2, dtype=torch.complex128)
+
+    return -0.5j * sum((L.mH @ L for L in lindbladians), start=zero)

emu_base/math/double_krylov.py

@@ -73,7 +73,9 @@ def lanczos(
     """
     converged = False
     lanczos_vectors = [v / v.norm()]
-    T = torch.zeros(max_krylov_dim + 2, max_krylov_dim + 2, dtype=v.dtype)
+    T = torch.zeros(
+        max_krylov_dim + 2, max_krylov_dim + 2, dtype=v.dtype, device=v.device
+    )
 
     for j in range(max_krylov_dim):
         w = op(lanczos_vectors[-1])

emu_base/pulser_adapter.py

@@ -3,12 +3,12 @@ from typing import Tuple, Sequence
 import torch
 import math
 from pulser.noise_model import NoiseModel
-from pulser.register.base_register import BaseRegister
+from pulser.register.base_register import BaseRegister, QubitId
 from enum import Enum
 
 from pulser.backend.config import EmulationConfig
 
-from emu_base.lindblad_operators import get_lindblad_operators
+from emu_base.jump_lindblad_operators import get_lindblad_operators
 from emu_base.utils import dist2, dist3
 
 
@@ -228,9 +228,11 @@ class PulserData:
     phi: torch.Tensor
     hamiltonian_type: HamiltonianType
     lindblad_ops: list[torch.Tensor]
+    qubit_ids: tuple[QubitId, ...]
 
     def __init__(self, *, sequence: pulser.Sequence, config: EmulationConfig, dt: int):
-        self.qubit_count = len(sequence.register.qubit_ids)
+        self.qubit_ids = sequence.register.qubit_ids
+        self.qubit_count = len(self.qubit_ids)
         sequence_duration = sequence.get_duration()
         # the end value is exclusive, so add +1
         observable_times = set(torch.arange(0, sequence.get_duration() + 1, dt).tolist())

{emu_base-2.0.3.dist-info → emu_base-2.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-base
-Version: 2.0.3
+Version: 2.1.0
 Summary: Pasqal base classes for emulators
 Project-URL: Documentation, https://pasqal-io.github.io/emulators/
 Project-URL: Repository, https://github.com/pasqal-io/emulators

emu_base-2.1.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+emu_base/__init__.py,sha256=nssAwoVeRB5OiBuBJcOzcfggDXp5IDB-2Xyft0kke6A,493
+emu_base/aggregators.py,sha256=bB-rldoDAErxQMpL715K5lpiabGOpkCY0GyxW7mfHuc,5000
+emu_base/constants.py,sha256=41LYkKLUCz-oxPbd-j7nUDZuhIbUrnez6prT0uR0jcE,56
+emu_base/jump_lindblad_operators.py,sha256=Y30f8emVFS4Dazljc_Rh4lX9qU4QQY_AxPNahnzcsfY,2101
+emu_base/pulser_adapter.py,sha256=jDNUpVDlcxfUgI3q5vGfB34YZI1prT39wf--HOwkOJA,11330
+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=X16dyCbyzdP7fFK-hmKS03Q-DJtC6TZ8sJrGTJ6akIc,3708
+emu_base/math/krylov_exp.py,sha256=MNLxgtiy2djRVtmXmtlBQ6A8rSuw1OK6dTtRQUZvaHs,3854
+emu_base-2.1.0.dist-info/METADATA,sha256=jkK_Y2sC5TRpYjxxTCouMtOklpiRu_EXSGpC6gkNp-s,3522
+emu_base-2.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_base-2.1.0.dist-info/RECORD,,

emu_base-2.0.3.dist-info/RECORD

@@ -1,13 +0,0 @@
-emu_base/__init__.py,sha256=FCku0o69e9ebQ5E2dYP2eVJeS9NC9O56AMEml0-vq7M,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
-emu_base/pulser_adapter.py,sha256=dRD80z_dVXkCjDBLRIkmqNGg5M78VEKkQuk3H5JdZSM,11241
-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=MNLxgtiy2djRVtmXmtlBQ6A8rSuw1OK6dTtRQUZvaHs,3854
-emu_base-2.0.3.dist-info/METADATA,sha256=vT_PZe1y41tVG2uljU4pz3HnSTBFzjVRk2swqQOx0ws,3522
-emu_base-2.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_base-2.0.3.dist-info/RECORD,,