emu-mps 2.0.3__py3-none-any.whl → 2.1.1__py3-none-any.whl

emu_mps/__init__.py

@@ -37,4 +37,4 @@ __all__ = [
     "EntanglementEntropy",
 ]
 
-__version__ = "2.0.3"
+__version__ = "2.1.1"

emu_mps/mps_backend_impl.py

@@ -24,7 +24,8 @@ from emu_mps.hamiltonian import make_H, update_H
 from emu_mps.mpo import MPO
 from emu_mps.mps import MPS
 from emu_mps.mps_config import MPSConfig
-from emu_mps.noise import compute_noise_from_lindbladians, pick_well_prepared_qubits
+from emu_mps.noise import pick_well_prepared_qubits
+from emu_base.jump_lindblad_operators import compute_noise_from_lindbladians
 import emu_mps.optimatrix as optimat
 from emu_mps.tdvp import (
     evolve_pair,
@@ -563,13 +564,18 @@ def permute_bitstrings(results: Results, perm: torch.Tensor) -> None:
 def permute_occupations_and_correlations(results: Results, perm: torch.Tensor) -> None:
     for corr in ["occupation", "correlation_matrix"]:
         if corr not in results.get_result_tags():
-            return
+            continue
 
         uuid_corr = results._find_uuid(corr)
         corrs = results._results[uuid_corr]
-        results._results[uuid_corr] = [
-            optimat.permute_tensor(corr, perm) for corr in corrs
-        ]
+        results._results[uuid_corr] = (
+            [  # vector quantities become lists after results are serialized (e.g. for checkpoints)
+                optimat.permute_tensor(
+                    corr if isinstance(corr, torch.Tensor) else torch.tensor(corr), perm
+                )
+                for corr in corrs
+            ]
+        )
 
 
 def permute_atom_order(results: Results, perm: torch.Tensor) -> None:

emu_mps/mps_config.py

@@ -43,8 +43,14 @@ class MPSConfig(EmulationConfig):
         num_gpus_to_use: during the simulation, distribute the state over this many GPUs
             0=all factors to cpu. As shown in the benchmarks, using multiple GPUs might
             alleviate memory pressure per GPU, but the runtime should be similar.
+        optimize_qubit_ordering: Optimize the register ordering. Improves performance and
+            accuracy, but disables certain features.
+        interaction_cutoff: Set interaction coefficients below this value to `0`.
+            Potentially improves runtime and memory consumption.
+        log_level: How much to log. Set to `logging.WARN` to get rid of the timestep info.
+        log_file: If specified, log to this file rather than stout.
         autosave_prefix: filename prefix for autosaving simulation state to file
-        autosave_dt: minimum time interval in seconds between two autosaves
+        autosave_dt: minimum time interval in seconds between two autosaves.
             Saving the simulation state is only possible at specific times,
             therefore this interval is only a lower bound.
         kwargs: arguments that are passed to the base class
@@ -206,7 +212,7 @@ class MPSConfig(EmulationConfig):
         not_allowed = actual_obs.difference(allowed_permutable_obs)
         if not_allowed:
             raise ValueError(
-                f"emu-mp allows only {allowed_permutable_obs} observables with"
+                f"emu-mps allows only {allowed_permutable_obs} observables with"
                 " `optimize_qubit_ordering = True`."
                 f" you provided unsupported {not_allowed}"
                 " To use other observables, please set"

emu_mps/noise.py

@@ -1,26 +1,6 @@
-import torch
 import random
 
 
-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"
-
-    return (
-        -1j
-        / 2.0
-        * sum(
-            (lindbladian.T.conj() @ lindbladian for lindbladian in lindbladians),
-            start=torch.zeros(2, 2, dtype=torch.complex128),
-        )
-    )
-
-
 def pick_well_prepared_qubits(eta: float, n: int) -> list[bool]:
     """
     Randomly pick n booleans such that ℙ(False) = eta.

{emu_mps-2.0.3.dist-info → emu_mps-2.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-mps
-Version: 2.0.3
+Version: 2.1.1
 Summary: Pasqal MPS 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.0.3
+Requires-Dist: emu-base==2.1.1
 Description-Content-Type: text/markdown
 
 <div align="center">

{emu_mps-2.0.3.dist-info → emu_mps-2.1.1.dist-info}/RECORD

@@ -1,19 +1,19 @@
-emu_mps/__init__.py,sha256=wGG0u1wHUcq_MuDjygyApshJY_ome-TZMVggIfHcgns,734
+emu_mps/__init__.py,sha256=8GuHi09MFuRHfvIoaIJ_S9wMOgLPw9_I52OO7Eosols,734
 emu_mps/algebra.py,sha256=ngPtTH-j2ZCBWoaJZXlkUyIlug7dY7Q92gzfnRlpPMA,5485
 emu_mps/custom_callback_implementations.py,sha256=CUs0kW3HRaPE7UeFNQOFbeWJMsz4hS2q4rgS57BBp-A,2411
 emu_mps/hamiltonian.py,sha256=gOPxNOBmk6jRPPjevERuCP_scGv0EKYeAJ0uxooihes,15622
 emu_mps/mpo.py,sha256=1ogQ25GZCwMzZ_m449oGHcYyDKrofBCr1eyzzrIPMhQ,8824
 emu_mps/mps.py,sha256=GIiWxctNmHARgf-PgQc6IHKNCe5HYSnbtlXI6Hc-0wI,20085
 emu_mps/mps_backend.py,sha256=bS83qFxvdoK-c12_1WaPw6O7xUc7vdWifZNHUzNP5sM,2091
-emu_mps/mps_backend_impl.py,sha256=8aPSn6ZGFf4DtTodzajX6WJtQHI9F8eMUN1AbrPzbgw,25611
-emu_mps/mps_config.py,sha256=JCK_frUVyMRJjI0NPU-dpCg-uO3JKeVzQMOLDkQ9XVk,8104
-emu_mps/noise.py,sha256=h4X2EFjoC_Ok0gZ8I9wN77RANXaVehTBbjkcbY_GAmY,784
+emu_mps/mps_backend_impl.py,sha256=U-fNHVkmQFwi_Hfun0OdJ0vmqi9ncjyJ4gIbEM1mN0Y,25887
+emu_mps/mps_config.py,sha256=WA64iI4SxxKRM8-49mjvXUzrUv4miYolVYOhR0mVmtk,8555
+emu_mps/noise.py,sha256=5BXthepWLKnuSTJfIFuPl2AcYPxUeTJdRc2b28ekkhg,208
 emu_mps/observables.py,sha256=7GQDH5kyaVNrwckk2f8ZJRV9Ca4jKhWWDsOCqYWsoEk,1349
 emu_mps/tdvp.py,sha256=0qTw9qhg0WbaAyBgeTpULHrNL0ytj80ZUb1P6GKD7Ww,6172
 emu_mps/utils.py,sha256=BqRJYAcXqprtZVJ0V_j954ON2bhTdtZiaTojsYyrWrg,8193
 emu_mps/optimatrix/__init__.py,sha256=fBXQ7-rgDro4hcaBijCGhx3J69W96qcw5_3mWc7tND4,364
 emu_mps/optimatrix/optimiser.py,sha256=k9suYmKLKlaZ7ozFuIqvXHyCBoCtGgkX1mpen9GOdOo,6977
 emu_mps/optimatrix/permutations.py,sha256=9DDMZtrGGZ01b9F3GkzHR3paX4qNtZiPoI7Z_Kia3Lc,3727
-emu_mps-2.0.3.dist-info/METADATA,sha256=ZU1USoiZ12em9lo4GgAWFiGE0oBT83yXitpm2fx8LvA,3505
-emu_mps-2.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_mps-2.0.3.dist-info/RECORD,,
+emu_mps-2.1.1.dist-info/METADATA,sha256=45NpjZK-7fd4dqVbo4Q76XqoMecHm_J2jiqX7oPUSKA,3505
+emu_mps-2.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_mps-2.1.1.dist-info/RECORD,,