emu-base 1.2.4__py3-none-any.whl → 1.2.6__py3-none-any.whl

emu_base/__init__.py

@@ -15,6 +15,7 @@ from .base_classes.default_callbacks import (
     StateResult,
     SecondMomentOfEnergy,
 )
+from .constants import DEVICE_COUNT
 from .pulser_adapter import PulserData, HamiltonianType
 from .math.brents_root_finding import find_root_brents
 from .math.krylov_exp import krylov_exp, DEFAULT_MAX_KRYLOV_DIM
@@ -42,6 +43,7 @@ __all__ = [
     "krylov_exp",
     "HamiltonianType",
     "DEFAULT_MAX_KRYLOV_DIM",
+    "DEVICE_COUNT",
 ]
 
-__version__ = "1.2.4"
+__version__ = "1.2.6"

emu_base/base_classes/config.py

@@ -4,6 +4,7 @@ import logging
 import sys
 import pathlib
 from typing import TYPE_CHECKING
+import torch
 
 if TYPE_CHECKING:
     from emu_base.base_classes.callback import Callback
@@ -50,14 +51,28 @@ class BackendConfig:
         self.with_modulation = with_modulation
         self.noise_model = noise_model
 
-        if interaction_matrix is not None and (
-            not isinstance(interaction_matrix, list)
-            or not isinstance(interaction_matrix[0], list)
+        if interaction_matrix is not None and not (
+            isinstance(interaction_matrix, list)
+            and isinstance(interaction_matrix[0], list)
+            and isinstance(interaction_matrix[0][0], float)
         ):
             raise ValueError(
                 "Interaction matrix must be provided as a Python list of lists of floats"
             )
 
+        if interaction_matrix is not None:
+            int_mat = torch.tensor(interaction_matrix)
+            tol = 1e-10
+            if not (
+                int_mat.numel() != 0
+                and torch.all(torch.isreal(int_mat))
+                and int_mat.dim() == 2
+                and int_mat.shape[0] == int_mat.shape[1]
+                and torch.allclose(int_mat, int_mat.T, atol=tol)
+                and torch.norm(torch.diag(int_mat)) < tol
+            ):
+                raise ValueError("Interaction matrix is not symmetric and zero diag")
+
         self.interaction_matrix = interaction_matrix
         self.interaction_cutoff = interaction_cutoff
         self.logger = logging.getLogger("global_logger")

emu_base/constants.py

@@ -0,0 +1,4 @@
+import torch
+
+
+DEVICE_COUNT = torch.cuda.device_count()

emu_base/pulser_adapter.py

@@ -158,17 +158,47 @@ def _extract_omega_delta_phi(
                 global_times |= set(i for i in range(slot.ti, slot.tf))
 
     step = 0
-    t = int((step + 1 / 2) * dt)
+    t = (step + 1 / 2) * dt
+    omega_1 = torch.zeros_like(omega[0])
+    omega_2 = torch.zeros_like(omega[0])
 
     while t < max_duration:
-        for q_pos, q_id in enumerate(sequence.register.qubit_ids):
-            omega[step, q_pos] = locals_a_d_p[q_id]["amp"][t]
-            delta[step, q_pos] = locals_a_d_p[q_id]["det"][t]
-            phi[step, q_pos] = locals_a_d_p[q_id]["phase"][t]
-        if t in global_times:
-            omega[step] *= waist_factors
+        # The sampled values correspond to the start of each interval
+        # To maximize the order of the solver, we need the values in the middle
+        if math.ceil(t) < max_duration:
+            # If we're not the final step, approximate this using linear interpolation
+            # Note that for dt even, t1=t2
+            for q_pos, q_id in enumerate(sequence.register.qubit_ids):
+                t1 = math.floor(t)
+                t2 = math.ceil(t)
+                omega_1[q_pos] = locals_a_d_p[q_id]["amp"][t1]
+                omega_2[q_pos] = locals_a_d_p[q_id]["amp"][t2]
+                delta[step, q_pos] = (
+                    locals_a_d_p[q_id]["det"][t1] + locals_a_d_p[q_id]["det"][t2]
+                ) / 2.0
+                phi[step, q_pos] = (
+                    locals_a_d_p[q_id]["phase"][t1] + locals_a_d_p[q_id]["phase"][t2]
+                ) / 2.0
+            # omegas at different times need to have the laser waist applied independently
+            if t1 in global_times:
+                omega_1 *= waist_factors
+            if t2 in global_times:
+                omega_2 *= waist_factors
+            omega[step] = 0.5 * (omega_1 + omega_2)
+        else:
+            # We're in the final step and dt=1, approximate this using linear extrapolation
+            # we can reuse omega_1 and omega_2 from before
+            for q_pos, q_id in enumerate(sequence.register.qubit_ids):
+                delta[step, q_pos] = (
+                    3.0 * locals_a_d_p[q_id]["det"][t2] - locals_a_d_p[q_id]["det"][t1]
+                ) / 2.0
+                phi[step, q_pos] = (
+                    3.0 * locals_a_d_p[q_id]["phase"][t2]
+                    - locals_a_d_p[q_id]["phase"][t1]
+                ) / 2.0
+            omega[step] = torch.clamp(0.5 * (3 * omega_2 - omega_1).real, min=0.0)
         step += 1
-        t = int((step + 1 / 2) * dt)
+        t = (step + 1 / 2) * dt
 
     return omega, delta, phi
 
@@ -241,11 +271,12 @@ class PulserData:
         ] = 0.0
         self.masked_interaction_matrix = self.full_interaction_matrix.clone()
 
-        slm_targets = sequence._slm_mask_targets
         self.slm_end_time = (
             sequence._slm_mask_time[1] if len(sequence._slm_mask_time) > 1 else 0.0
         )
 
-        for target in slm_targets:
+        # disable interaction for SLM masked qubits
+        slm_targets = list(sequence._slm_mask_targets)
+        for target in sequence.register.find_indices(slm_targets):
             self.masked_interaction_matrix[target] = 0.0
             self.masked_interaction_matrix[:, target] = 0.0

{emu_base-1.2.4.dist-info → emu_base-1.2.6.dist-info}/METADATA

@@ -1,7 +1,10 @@
 Metadata-Version: 2.4
 Name: emu-base
-Version: 1.2.4
+Version: 1.2.6
 Summary: Pasqal base classes for emulators
+Project-URL: Documentation, https://pasqal-io.github.io/emulators/
+Project-URL: Repository, https://github.com/pasqal-io/emulators
+Project-URL: Issues, https://github.com/pasqal-io/emulators/issues
 Author-email: Anton Quelle <anton.quelle@pasqal.com>, Mauro Mendizabal <mauro.mendizabal-pico@pasqal.com>, Stefano Grava <stefano.grava@pasqal.com>, Pablo Le Henaff <pablo.le-henaff@pasqal.com>
 License: PASQAL OPEN-SOURCE SOFTWARE LICENSE AGREEMENT (MIT-derived)
         
@@ -22,79 +25,14 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.10
-Requires-Dist: pulser-core==1.1.*
+Requires-Dist: pulser-core==1.3.*
 Requires-Dist: torch==2.5.0
 Description-Content-Type: text/markdown
 
 <div align="center">
   <img src="docs/logos/LogoTaglineSoftGreen.svg">
-
-  # Emu-MPS
 </div>
 
-**Emu-mps** is a backend for the [Pulser low-level Quantum Programming toolkit](https://pulser.readthedocs.io) that lets you run Quantum Algorithms on a simulated device, using GPU acceleration if available. More in depth, emu-mps is designed to **emu**late the dynamics of programmable arrays of neutral atoms, with matrix product states (**mps**). While benchmarking is incomplete as of this writing, early results suggest that this design makes emu-mps faster and more memory-efficient than previous generations of quantum emulators at running simulations with large numbers of qubits.
-
-## Installation
-
-**Warning:** installing emu-mps will update pulser-core
-
-### Using `hatch`, `uv` or any pyproject-compatible Python manager
-
-To add `emu-mps` to your project, edit your `pyproject.toml` to add the line
-
-```toml
-  "emu-mps"
-```
-
-to the list of `dependencies`.
-
-
-### Using `pip` or `pipx`
-To install the `pipy` package using `pip` or `pipx`
-
-1. Create a `venv` if that's not done yet
-
-```sh
-$ python -m venv venv
-
-```
-
-2. Enter the venv
-
-If you're running Unix:
-
-```sh
-$ . venv/bin/activate
-```
-
-If you're running Windows:
-
-```sh
-C:\> /path/to/new/virtual/environment/Scripts/activate
-```
-
-3. Install the package
-
-```sh
-$ pip install emu-mps
-# or
-$ pipx install emu-mps
-```
-
-
-Join us on [Slack](https://pasqalworkspace.slack.com/archives/C07MUV5K7EU) or by [e-mail](mailto:emulation@pasqal.com) to give us feedback about how you plan to use Emu-MPS or if you require specific feature-upgrades.
-
-## Usage
-
-For the time being, the easiest way to learn how to use this package is to look
-at the [examples](examples/emu_mps_examples) and [notebooks](https://pasqal-io.github.io/emulators/latest/).
-
-See also the [full documentation](https://github.com/pasqal-io/emulators/blob/main/docs/index.md) for
-the API, information about contributing, benchmarks, etc.
-
-
-## Getting in touch
+# Welcome to the Pasqal analog emulators
 
-- [Pasqal Community Portal](https://community.pasqal.com/) (forums, chat, tutorials, examples, code library).
-- [GitHub Repository](https://github.com/pasqal-io/quantum-evolution-kernel) (source code, issue tracker).
-- [Professional Support](https://www.pasqal.com/contact-us/) (if you need tech support, custom licenses, a variant of this library optimized for your workload, your own QPU, remote access to a QPU, ...)
+Welcome, and please see the [GitHub pages](https://pasqal-io.github.io/emulators/) for a landing page to this repo.

{emu_base-1.2.4.dist-info → emu_base-1.2.6.dist-info}/RECORD

@@ -1,12 +1,13 @@
-emu_base/__init__.py,sha256=7xlIJ_5eU7fEVG-RQnIdSnCrRk2fw5NwqjYmwmUwKC8,1130
+emu_base/__init__.py,sha256=YDB_UjuJJoFfzL3gG_z09ieMTU7RvE5UAGn7POqiYbE,1186
+emu_base/constants.py,sha256=41LYkKLUCz-oxPbd-j7nUDZuhIbUrnez6prT0uR0jcE,56
 emu_base/lindblad_operators.py,sha256=eXkXsLt_fV7jhF31EkxzYFU04rOJV3uWXsl6t1aTAqs,1562
-emu_base/pulser_adapter.py,sha256=aFq4Q7INUTyhfqALSrLpZZRnKf1QZ0SyUyt1Cdiolz0,8802
+emu_base/pulser_adapter.py,sha256=Doi24xNQp09UhyrvNkS_ydXiRV4XNinoziALo6Gil2U,10456
 emu_base/utils.py,sha256=RM8O0qfPAJfcdqqAojwEEKV7I3ZfVDklnTisTGhUg5k,233
 emu_base/base_classes/__init__.py,sha256=Su6fHtjCyg0fw-7y7e7nbMfDASppNRQs8iGaAOkO3c4,570
 emu_base/base_classes/aggregators.py,sha256=bcvoGfZCkPKv-CI29gTma6HBphGh7bjBq2Ome77eRJM,1840
 emu_base/base_classes/backend.py,sha256=7tnwb9MnRbwRN1_JTqliYftjqExuOE-Rrwz9AU2Pc4c,1645
 emu_base/base_classes/callback.py,sha256=JXah_ZDNM8iyPWy7IOwW481qRFyqVvlSM-0OkjBzV0A,3055
-emu_base/base_classes/config.py,sha256=Iz5bO6WRcjiz3ORGscyM7ELQZ4IocuJLJjJLyc-Odpg,3474
+emu_base/base_classes/config.py,sha256=Dg2CwC9sc5HYwszQAJSVjkSd3wuQrv6aEZGYBRnFl48,4098
 emu_base/base_classes/default_callbacks.py,sha256=F44kkuwWdVcvMGZ9vJ2q7ug-_P8IQyJv-SVxSVWHW_w,9940
 emu_base/base_classes/operator.py,sha256=MJjuDUTwJLbaSJzSNCKDWGvmGCGAEIEWISLoPSSzNsU,3501
 emu_base/base_classes/results.py,sha256=7-Mz3jmFy19hd3PIA5idK610mC3b5jOf3EKBmV14Jv4,5569
@@ -14,6 +15,6 @@ emu_base/base_classes/state.py,sha256=7iIyZmBqqJ6G4SyYZ3kyylWjAqiYIx0aW5B0T74EPZ
 emu_base/math/__init__.py,sha256=6BbIytYV5uC-e5jLMtIErkcUl_PvfSNnhmVFY9Il8uQ,97
 emu_base/math/brents_root_finding.py,sha256=AVx6L1Il6rpPJWrLJ7cn6oNmJyZOPRgEaaZaubC9lsU,3711
 emu_base/math/krylov_exp.py,sha256=UCFNeq-j2ukgBsOPC9_Jiv1aqpy88SrslDLiCxIGBwk,3840
-emu_base-1.2.4.dist-info/METADATA,sha256=uNaPv60Dih8BsidRdd1bEEfBTJGstIC-BuxjkbUWaqQ,5482
-emu_base-1.2.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_base-1.2.4.dist-info/RECORD,,
+emu_base-1.2.6.dist-info/METADATA,sha256=AKKbxP9g8y0vjIYHix98y-DxKOlAufhKRW8YQy74h_Q,3522
+emu_base-1.2.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_base-1.2.6.dist-info/RECORD,,