emu-base 2.4.1__py3-none-any.whl → 2.4.3__py3-none-any.whl

emu_base/__init__.py

@@ -21,4 +21,4 @@ __all__ = [
     "unix_like",
 ]
 
-__version__ = "2.4.1"
+__version__ = "2.4.3"

emu_base/jump_lindblad_operators.py

@@ -2,43 +2,59 @@ from pulser.noise_model import NoiseModel
 import torch
 import math
 
+dtype = torch.complex128
+
 
 def get_lindblad_operators(
-    *, noise_type: str, noise_model: NoiseModel
+    *, noise_type: str, noise_model: NoiseModel, dim: int = 2
 ) -> list[torch.Tensor]:
+
     assert noise_type in noise_model.noise_types
 
     if noise_type == "relaxation":
         c = math.sqrt(noise_model.relaxation_rate)
-        return [
-            torch.tensor([[0, c], [0, 0]], dtype=torch.complex128),
-        ]
+        relaxation = torch.zeros(dim, dim, dtype=dtype)
+        relaxation[0, 1] = c
+        return [relaxation]
 
     if noise_type == "dephasing":
         if noise_model.hyperfine_dephasing_rate != 0.0:
             raise NotImplementedError("hyperfine_dephasing_rate is unsupported")
 
         c = math.sqrt(noise_model.dephasing_rate / 2)
-        return [
-            torch.tensor([[-c, 0], [0, c]], dtype=torch.complex128),
-        ]
+        dephasing = torch.zeros(dim, dim, dtype=dtype)
+
+        dephasing[0, 0] = c
+        dephasing[1, 1] = -c
+
+        return [dephasing]
 
     if noise_type == "depolarizing":
         c = math.sqrt(noise_model.depolarizing_rate / 4)
-        return [
-            torch.tensor([[0, c], [c, 0]], dtype=torch.complex128),
-            torch.tensor([[0, 1j * c], [-1j * c, 0]], dtype=torch.complex128),
-            torch.tensor([[-c, 0], [0, c]], dtype=torch.complex128),
-        ]
+        depolarizing_x = torch.zeros(dim, dim, dtype=dtype)
+        depolarizing_x[0, 1] = c
+        depolarizing_x[1, 0] = c
+
+        depolarizing_y = torch.zeros(dim, dim, dtype=dtype)
+        depolarizing_y[0, 1] = torch.tensor(-c * 1.0j, dtype=dtype)
+        depolarizing_y[1, 0] = torch.tensor(c * 1.0j, dtype=dtype)
+
+        depolarizing_z = torch.zeros(dim, dim, dtype=dtype)
+        depolarizing_z[0, 0] = c
+        depolarizing_z[1, 1] = -c
+
+        return [depolarizing_x, depolarizing_y, depolarizing_z]
 
     if noise_type == "eff_noise":
         if not all(
-            isinstance(op, torch.Tensor) and op.shape == (2, 2)
+            isinstance(op, torch.Tensor) and op.shape == (dim, dim)
             for op in noise_model.eff_noise_opers
         ):
-            raise ValueError("Only 2 * 2 effective noise operator matrices are supported")
+            raise ValueError(
+                "Only 2 by 2 or 3 by 3 effective noise operator matrices are supported"
+            )
 
-        return [
+        return [  # lindblad operators are coming from pulser
             math.sqrt(rate)
             * torch.flip(op if isinstance(op, torch.Tensor) else torch.tensor(op), (0, 1))
             for rate, op in zip(noise_model.eff_noise_rates, noise_model.eff_noise_opers)
@@ -47,14 +63,17 @@ def get_lindblad_operators(
     raise ValueError(f"Unknown noise type: {noise_type}")
 
 
-def compute_noise_from_lindbladians(lindbladians: list[torch.Tensor]) -> torch.Tensor:
+def compute_noise_from_lindbladians(
+    lindbladians: list[torch.Tensor], dim: int = 2
+) -> torch.Tensor:
     """
-    Compute the single-qubit Hamiltonian noise term -0.5i∑L†L from all the given lindbladians.
+    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
+        lindbladian.shape == (dim, dim) for lindbladian in lindbladians
     ), "Only single-qubit lindblad operators are supported"
 
-    zero = torch.zeros(2, 2, dtype=torch.complex128)
+    zero = torch.zeros(dim, dim, dtype=dtype)
 
     return -0.5j * sum((L.mH @ L for L in lindbladians), start=zero)

emu_base/pulser_adapter.py

@@ -20,7 +20,7 @@ _NON_LINDBLADIAN_NOISE = {"SPAM", "doppler", "amplitude", "detuning", "register"
 
 
 def _get_all_lindblad_noise_operators(
-    noise_model: NoiseModel | None,
+    noise_model: NoiseModel | None, dim: int = 2
 ) -> list[torch.Tensor]:
     if noise_model is None:
         return []
@@ -29,7 +29,9 @@ def _get_all_lindblad_noise_operators(
         op
         for noise_type in noise_model.noise_types
         if noise_type not in _NON_LINDBLADIAN_NOISE
-        for op in get_lindblad_operators(noise_type=noise_type, noise_model=noise_model)
+        for op in get_lindblad_operators(
+            noise_type=noise_type, noise_model=noise_model, dim=dim
+        )
     ]
 
 
@@ -150,6 +152,7 @@ class PulserData:
         )
 
         int_type = self.hamiltonian.interaction_type
+        self.dim = self.hamiltonian.dim
         if int_type == "ising":  # for local and global
             self.hamiltonian_type = HamiltonianType.Rydberg
         elif int_type == "XY":
@@ -157,7 +160,9 @@ class PulserData:
         else:
             raise ValueError(f"Unsupported basis: {int_type}")
 
-        self.lindblad_ops = _get_all_lindblad_noise_operators(config.noise_model)
+        self.lindblad_ops = _get_all_lindblad_noise_operators(
+            config.noise_model, dim=self.dim
+        )
         self.has_lindblad_noise: bool = self.lindblad_ops != []
 
         if config.interaction_matrix is not None:

{emu_base-2.4.1.dist-info → emu_base-2.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-base
-Version: 2.4.1
+Version: 2.4.3
 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.6.*
-Requires-Dist: torch>=2.8.0
+Requires-Dist: torch>=2.9.0
 Description-Content-Type: text/markdown
 
 <div align="center">

{emu_base-2.4.1.dist-info → emu_base-2.4.3.dist-info}/RECORD

@@ -1,7 +1,7 @@
-emu_base/__init__.py,sha256=hxXLbvqH2yhva3QXRtTws6oSFnVPMndrgvENXzze9Rw,725
+emu_base/__init__.py,sha256=2_F4En9Wj4MwnJXavLN7LqV5CnOnV0JGSCFBSCuGhsw,725
 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=bb529P2tspLgX_EcJxCT4bJjFM8SOIttgsFIznTeN58,6979
+emu_base/jump_lindblad_operators.py,sha256=b5xL62aCVFjbNRigyBtRTacG4Xg2A3eYXRe2KBNlt9w,2588
+emu_base/pulser_adapter.py,sha256=NB220g-ESyi8Te-mcmj0ciLpi4JG-Bj7dVKy5sKJuUU,7099
 emu_base/utils.py,sha256=OoJ0GjL1K8m9Jq-BeBfmbWuFp9bvwXp8UcI3BMW94Js,2974
 emu_base/math/__init__.py,sha256=6BbIytYV5uC-e5jLMtIErkcUl_PvfSNnhmVFY9Il8uQ,97
 emu_base/math/brents_root_finding.py,sha256=AVx6L1Il6rpPJWrLJ7cn6oNmJyZOPRgEaaZaubC9lsU,3711
@@ -9,6 +9,6 @@ emu_base/math/double_krylov.py,sha256=X16dyCbyzdP7fFK-hmKS03Q-DJtC6TZ8sJrGTJ6akI
 emu_base/math/krylov_energy_min.py,sha256=iR4hmE0eXptbAg3opikd5d4Zv7dhnDrawH-n_4KG-cc,4009
 emu_base/math/krylov_exp.py,sha256=mGFddVQ8mEbwypbZtnlRPFpi4Nf8JZT6OKLHloIwCDQ,3934
 emu_base/math/matmul.py,sha256=lEAnV0b5z_f1xEA-9p-WXxA8bM3QbShiHdXQ3ZkZFcQ,877
-emu_base-2.4.1.dist-info/METADATA,sha256=8ikaKvc4pjqEikcqgoCUj4WnRUlClI5xBQfIPDRuFO8,3604
-emu_base-2.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_base-2.4.1.dist-info/RECORD,,
+emu_base-2.4.3.dist-info/METADATA,sha256=atbuyLSrlYZNsVpLubWaBV4yblkgllwW45DvmFKvZlE,3604
+emu_base-2.4.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_base-2.4.3.dist-info/RECORD,,