emu-sv 2.2.0__py3-none-any.whl → 2.2.1__py3-none-any.whl

emu_sv/__init__.py

@@ -38,4 +38,4 @@ __all__ = [
     "DensityMatrix",
 ]
 
-__version__ = "2.2.0"
+__version__ = "2.2.1"

emu_sv/custom_callback_implementations.py

@@ -6,12 +6,12 @@ from pulser.backend import (
     EnergySecondMoment,
     EnergyVariance,
     Occupation,
-    Energy,
 )
 from emu_sv.density_matrix_state import DensityMatrix
 from emu_sv.state_vector import StateVector
 from emu_sv.dense_operator import DenseOperator
 from emu_sv.hamiltonian import RydbergHamiltonian
+from emu_sv.lindblad_operator import RydbergLindbladian
 
 dtype = torch.float64
 
@@ -134,6 +134,26 @@ def energy_variance_sv_impl(
     return en_var.cpu()
 
 
+def energy_variance_sv_den_mat_impl(
+    self: EnergyVariance,
+    *,
+    config: EmulationConfig,
+    state: DensityMatrix,
+    hamiltonian: RydbergLindbladian,
+) -> torch.Tensor:
+    """
+    Custom implementation of the energy variance tr(ρH²)-tr(ρH)² for the
+    lindblad equation solver.
+    """
+    h_dense_matrix = hamiltonian.h_eff(state.matrix)  # Hρ
+    gpu = state.matrix.is_cuda
+    h_squared_dense_mat = hamiltonian.expect(
+        DensityMatrix(h_dense_matrix, gpu=gpu)
+    )  # tr(ρH²)
+    en_var: torch.Tensor = h_squared_dense_mat - hamiltonian.expect(state) ** 2  # tr(ρH)²
+    return en_var.cpu()
+
+
 def energy_second_moment_sv_impl(
     self: EnergySecondMoment,
     *,
@@ -150,15 +170,20 @@ def energy_second_moment_sv_impl(
     return en_2_mom.cpu()
 
 
-def energy_sv_impl(
-    self: Energy,
+def energy_second_moment_den_mat_impl(
+    self: EnergyVariance,
     *,
     config: EmulationConfig,
-    state: StateVector,
-    hamiltonian: RydbergHamiltonian,
+    state: DensityMatrix,
+    hamiltonian: RydbergLindbladian,
 ) -> torch.Tensor:
     """
-    Custom implementation of the energy ❬ψ|H|ψ❭ for the state vector solver.
+    Custom implementation of the second moment of energy tr(ρH²) for the
+    lindblad equation solver.
     """
-    en: torch.Tensor = hamiltonian.expect(state)
-    return en
+    h_dense_matrix = hamiltonian.h_eff(state.matrix)  # Hρ
+    gpu = state.matrix.is_cuda
+
+    return hamiltonian.expect(
+        DensityMatrix(h_dense_matrix, gpu=gpu)
+    ).cpu()  # tr(ρH²) = tr(ρ H H)

emu_sv/lindblad_operator.py

@@ -1,6 +1,6 @@
 import torch
 from emu_base import compute_noise_from_lindbladians, matmul_2x2_with_batched
-
+from emu_sv.density_matrix_state import DensityMatrix
 
 dtype = torch.complex128
 sigmax = torch.tensor([[0.0, 1.0], [1.0, 0.0]], dtype=dtype)
@@ -129,6 +129,52 @@ class RydbergLindbladian:
 
         return density_matrix.view(orignal_shape)
 
+    def h_eff(
+        self,
+        density_matrix: torch.Tensor,
+        lindblad_ops: torch.Tensor = torch.zeros(2, 2, dtype=dtype),
+    ) -> torch.Tensor:
+        """Compute the effective Hamiltonian, Heff = Hρ  -0.5i ∑ₖ Lₖ† Lₖ ρ, applied
+        to a density matrix ρ.
+        - libdlad_ops by default are 2x2 zero matrix"""
+        H_den_matrix = torch.zeros_like(density_matrix, dtype=dtype, device=self.device)
+
+        for qubit in range(len(self.omegas)):
+            H_q = self._local_terms_hamiltonian(qubit, lindblad_ops.to(self.device))
+            H_den_matrix += self.apply_local_op_to_density_matrix(
+                density_matrix, H_q, qubit
+            )
+
+        H_den_matrix += self._apply_interaction_terms(density_matrix)
+        return H_den_matrix
+
+    def _local_terms_hamiltonian(
+        self, qubit: int, lindblad_ops: torch.Tensor
+    ) -> torch.Tensor:
+        """Construct the Hamiltonian for single qubit terms:
+        ∑ᵢ 𝛺ᵢ /2 𝜎ₓ^ i − 𝛿ⁱ nᵢ + jump operators terms , including the sum
+        of  Lindblad terms, when 𝜙ᵢ equals to 0.0.
+        When 𝜙ᵢ not equals to 0.0:
+        ∑ᵢ 𝛺ᵢ /2 (cos (𝜙ᵢ) 𝜎ₓ^ i + sin(𝜙ᵢ)𝜎_yⁱ) − 𝛿ⁱ nᵢ + jump operators terms
+        """
+        omega = self.omegas[qubit]
+        delta = self.deltas[qubit]
+
+        sigma_x = sigmax.to(device=self.device)
+        n = n_op.to(device=self.device)
+
+        if not self.complex:
+            return omega * sigma_x - delta * n + lindblad_ops.to(self.device)
+
+        phi = self.phis[qubit]
+        sigma_y = sigmay.to(device=self.device)
+        complex_part = torch.cos(phi) * sigma_x + torch.sin(phi) * sigma_y
+        return omega * complex_part - delta * n + lindblad_ops.to(self.device)
+
+    def _apply_interaction_terms(self, density_matrix: torch.Tensor) -> torch.Tensor:
+        """Apply the interaction terms ∑ᵢⱼ Uᵢⱼ nᵢ nⱼ to the density matrix."""
+        return self.diag.view(-1, 1) * density_matrix
+
     def __matmul__(self, density_matrix: torch.Tensor) -> torch.Tensor:
         """Apply the i*RydbergLindbladian operator to the density matrix ρ
         in the following way:
@@ -142,41 +188,8 @@ class RydbergLindbladian:
         sum_lindblad_local = compute_noise_from_lindbladians(self.pulser_linblads).to(
             self.device
         )
-
-        # apply all local terms:  Ωⱼ σₓ - δⱼ n - 0.5i (∑ₖ Lₖ† Lₖ) to each qubit
-        H_den_matrix = torch.zeros_like(density_matrix, dtype=dtype, device=self.device)
-
-        if not self.complex:
-            for qubit, (omega, delta) in enumerate(zip(self.omegas, self.deltas)):
-                H_q = (
-                    omega * sigmax.to(device=self.device)
-                    - delta * n_op.to(device=self.device)
-                    + sum_lindblad_local
-                )
-                H_den_matrix += self.apply_local_op_to_density_matrix(
-                    density_matrix, H_q, qubit
-                )
-        else:
-            for qubit, (omega, delta, phi) in enumerate(
-                zip(self.omegas, self.deltas, self.phis)
-            ):
-                H_q = (
-                    omega
-                    * (
-                        (
-                            torch.cos(phi) * sigmax.to(device=self.device)
-                            + torch.sin(phi) * sigmay.to(device=self.device)
-                        )
-                    )
-                    - delta * n_op.to(device=self.device)
-                    + sum_lindblad_local
-                )
-                H_den_matrix += self.apply_local_op_to_density_matrix(
-                    density_matrix, H_q, qubit
-                )
-
-        # apply the interaction terms  ∑ᵢⱼ Uᵢⱼ nᵢ nⱼ
-        H_den_matrix += self.diag.view(-1, 1) * density_matrix
+        # Heff = Hρ  -0.5i ∑ₖ Lₖ† Lₖ ρ
+        H_den_matrix = self.h_eff(density_matrix, sum_lindblad_local)
 
         # Heff - Heff^†=  [H, ρ] - 0.5i ∑ₖ Lₖ† Lₖρ - ρ 0.5i ∑ₖ Lₖ† Lₖρ
         H_den_matrix = H_den_matrix - H_den_matrix.conj().T
@@ -195,3 +208,10 @@ class RydbergLindbladian:
         )
 
         return H_den_matrix + 1.0j * L_den_matrix_Ldag
+
+    def expect(self, state: DensityMatrix) -> torch.Tensor:
+        """Return the energy expectation value E=tr(H𝜌)"""
+        en = (self.h_eff(state.matrix)).trace()
+
+        assert torch.allclose(en.imag, torch.zeros_like(en.imag), atol=1e-8)
+        return en.real

emu_sv/sv_config.py

@@ -9,7 +9,9 @@ from emu_sv.custom_callback_implementations import (
     correlation_matrix_sv_impl,
     correlation_matrix_sv_den_mat_impl,
     energy_second_moment_sv_impl,
+    energy_second_moment_den_mat_impl,
     energy_variance_sv_impl,
+    energy_variance_sv_den_mat_impl,
     qubit_occupation_sv_impl,
     qubit_occupation_sv_den_mat_impl,
 )
@@ -144,17 +146,22 @@ class SVConfig(EmulationConfig):
                     obs_copy,
                 )
             if isinstance(obs, EnergyVariance):
-                if self.noise_model.noise_types != ():
-                    raise Exception("Not implemented for density matrix")
                 obs_copy.apply = MethodType(  # type: ignore[method-assign]
-                    energy_variance_sv_impl, obs_copy
+                    (
+                        energy_variance_sv_impl
+                        if self.noise_model.noise_types == ()
+                        else energy_variance_sv_den_mat_impl
+                    ),
+                    obs_copy,
                 )
             elif isinstance(obs, EnergySecondMoment):
-                if self.noise_model.noise_types != ():
-                    raise Exception("Not implemented for density matrix")
-
                 obs_copy.apply = MethodType(  # type: ignore[method-assign]
-                    energy_second_moment_sv_impl, obs_copy
+                    (
+                        energy_second_moment_sv_impl
+                        if self.noise_model.noise_types == ()
+                        else energy_second_moment_den_mat_impl
+                    ),
+                    obs_copy,
                 )
             obs_list.append(obs_copy)
         self.observables = tuple(obs_list)

{emu_sv-2.2.0.dist-info → emu_sv-2.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-sv
-Version: 2.2.0
+Version: 2.2.1
 Summary: Pasqal State Vector based pulse emulator built on PyTorch
 Project-URL: Documentation, https://pasqal-io.github.io/emulators/
 Project-URL: Repository, https://github.com/pasqal-io/emulators
@@ -25,7 +25,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.10
-Requires-Dist: emu-base==2.2.0
+Requires-Dist: emu-base==2.2.1
 Description-Content-Type: text/markdown
 
 <div align="center">

{emu_sv-2.2.0.dist-info → emu_sv-2.2.1.dist-info}/RECORD

@@ -1,15 +1,15 @@
-emu_sv/__init__.py,sha256=vVU7zIUsHPgFXUBWMn1lUtkWUg-avdFRqLUrtbmPvDI,771
-emu_sv/custom_callback_implementations.py,sha256=j_G5x-xSnFSyKE81QN_DKoiTaF9JvNWaM1N6k2_t0Oo,5501
+emu_sv/__init__.py,sha256=sREepAk-h3a3-e7PGexVNmYmj5hf4fVIrKk78vnDKiE,771
+emu_sv/custom_callback_implementations.py,sha256=_7XLIDzJ-p3DVqz-Jyv0eYbl8nih2x2p-pM4cBCLumA,6367
 emu_sv/dense_operator.py,sha256=NfgzVpnNitc5ZSM4RlfpAc5Ls2wFPNsTxdeFdhJSg1o,6909
 emu_sv/density_matrix_state.py,sha256=5W_UKIAYHb0k3ryRLQ2dbFUgrb5ju5jceDGAekM2gNE,7035
 emu_sv/hamiltonian.py,sha256=CqNGuWJlO2ZljK47wt130s-5uKiOldQUsC3tjwk1mKA,6106
-emu_sv/lindblad_operator.py,sha256=Rlxh24dXVUAutSrSacNO2ilNVlxKix8pfFt7h2CfVOg,7893
+emu_sv/lindblad_operator.py,sha256=pgjRNLBcvEM2-qxM8uy9wL74OtrD4A8trQeERi_AXH8,8892
 emu_sv/state_vector.py,sha256=zKHCdgl_eRIOPE4qVKO53ig9UyYTQ7a_guNFXgynU7g,9753
 emu_sv/sv_backend.py,sha256=-soOkSEzEBK1dCKnYnbtvYjmNZtZra1_4jP3H1ROOtM,737
 emu_sv/sv_backend_impl.py,sha256=mdPWBLDwH0q7EEwQTmLNLLx5tycMmsCQbUifIHvciMk,8059
-emu_sv/sv_config.py,sha256=ixMTgDXKll4bXsYtAe4a_9Gng0bhwCgS42KKMwZCFHI,6308
+emu_sv/sv_config.py,sha256=0geciKkrF3h8pU_UAQ8R-G6WxfYb_X5XIrxZavGxK5Q,6511
 emu_sv/time_evolution.py,sha256=_VH4f2RF6lGKzO08WxTYJ5Aw8_pTTMRKcyMnIuxH03I,13382
 emu_sv/utils.py,sha256=-axfQ2tqw0C7I9yw-28g7lytyk373DNBjDALh4kLBrM,302
-emu_sv-2.2.0.dist-info/METADATA,sha256=aw71GDNhy2P9-zjXjXszkLyaOY95qCpPkOmG2lAmYZg,3595
-emu_sv-2.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_sv-2.2.0.dist-info/RECORD,,
+emu_sv-2.2.1.dist-info/METADATA,sha256=_NvCSEGYdw9ZXATsZKME-GX0ECeiJcDy8MthY9C5ZUU,3595
+emu_sv-2.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_sv-2.2.1.dist-info/RECORD,,