qadence 1.5.0__tar.gz → 1.5.2__tar.gz

{qadence-1.5.0 → qadence-1.5.2}/.pre-commit-config.yaml

@@ -20,7 +20,7 @@ repos:
       - id: black
 
 -   repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.7.0
+    rev: v1.9.0
     hooks:
       - id: mypy
         exclude: examples|docs

{qadence-1.5.0 → qadence-1.5.2}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.3
 Name: qadence
-Version: 1.5.0
+Version: 1.5.2
 Summary: Pasqal interface for circuit-based quantum computing SDKs
-Author-email: Aleksander Wennersteen <aleksander.wennersteen@pasqal.com>, Gert-Jan Both <gert-jan.both@pasqal.com>, Niklas Heim <niklas.heim@pasqal.com>, Mario Dagrada <mario.dagrada@pasqal.com>, Vincent Elfving <vincent.elfving@pasqal.com>, Dominik Seitz <dominik.seitz@pasqal.com>, Roland Guichard <roland.guichard@pasqal.com>, "Joao P. Moutinho" <joao.moutinho@pasqal.com>, Vytautas Abramavicius <vytautas.abramavicius@pasqal.com>, Gergana Velikova <gergana.velikova@pasqal.com>
+Author-email: Aleksander Wennersteen <aleksander.wennersteen@pasqal.com>, Gert-Jan Both <gert-jan.both@pasqal.com>, Niklas Heim <niklas.heim@pasqal.com>, Mario Dagrada <mario.dagrada@pasqal.com>, Vincent Elfving <vincent.elfving@pasqal.com>, Dominik Seitz <dominik.seitz@pasqal.com>, Roland Guichard <roland.guichard@pasqal.com>, "Joao P. Moutinho" <joao.moutinho@pasqal.com>, Vytautas Abramavicius <vytautas.abramavicius@pasqal.com>, Gergana Velikova <gergana.velikova@pasqal.com>, Eduardo Maschio <eduardo.maschio@pasqal.com>
 License: Apache 2.0
 License-File: LICENSE
 Classifier: License :: OSI Approved :: Apache Software License
@@ -14,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: <3.13,>=3.9
+Requires-Dist: arpeggio==2.0.2
 Requires-Dist: deepdiff
 Requires-Dist: jsonschema
 Requires-Dist: matplotlib
@@ -27,7 +28,7 @@ Requires-Dist: sympytorch>=0.1.2
 Requires-Dist: tensorboard>=2.12.0
 Requires-Dist: torch
 Provides-Extra: all
-Requires-Dist: amazon-braket-sdk; extra == 'all'
+Requires-Dist: amazon-braket-sdk==1.71.0; extra == 'all'
 Requires-Dist: graphviz; extra == 'all'
 Requires-Dist: libs; extra == 'all'
 Requires-Dist: protocols; extra == 'all'
@@ -53,19 +54,21 @@ Provides-Extra: visualization
 Requires-Dist: graphviz; extra == 'visualization'
 Description-Content-Type: text/markdown
 
-<picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/assets/logo/qadence_logo_white.svg">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/assets/logo/qadence_logo.svg">
-  <img alt="Qadence logo" src="./docs/assets/logo/qadence_logo.svg">
-</picture>
+<p align="center">
+  <picture>
+    <source media="(prefers-color-scheme: dark)" srcset="./docs/extras/assets/logo/qadence_logo_white.svg" width="75%">
+    <source media="(prefers-color-scheme: light)" srcset="./docs/extras/assets/logo/qadence_logo.svg" width="75%">
+    <img alt="Qadence logo" src="./docs/assets/logo/qadence_logo.svg" width="75%">
+  </picture>
+</p>
+
+**Qadence** is a Python package that provides a simple interface to build **digital-analog quantum
+programs** with tunable qubit interactions and arbitrary register topologies realizable on neutral atom devices.
 
 **For a high-level overview of Qadence features, [check out our white paper](https://arxiv.org/abs/2401.09915).**
 
 **For more detailed information, [check out the documentation](https://pasqal-io.github.io/qadence/latest/).**
 
-**Qadence** is a Python package that provides a simple interface to build _**digital-analog quantum
-programs**_ with tunable qubit interaction defined on _**arbitrary register topologies**_ realizable on neutral atom devices.
-
 [![Linting](https://github.com/pasqal-io/qadence/actions/workflows/lint.yml/badge.svg)](https://github.com/pasqal-io/qadence/actions/workflows/lint.yml)
 [![Tests](https://github.com/pasqal-io/qadence/actions/workflows/test_fast.yml/badge.svg)](https://github.com/pasqal-io/qadence/actions/workflows/test_fast.yml)
 [![Documentation](https://github.com/pasqal-io/qadence/actions/workflows/build_docs.yml/badge.svg)](https://pasqal-io.github.io/qadence/latest)
@@ -74,27 +77,26 @@ programs**_ with tunable qubit interaction defined on _**arbitrary register topo
 
 ## Feature highlights
 
-<picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/assets/qadence_arch.svg">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/assets/qadence_arch.svg">
-  <img alt="Qadence architecture" src="./docs/assets/qadence_arch.svg">
-</picture>
-
+<p align="center">
+  <picture>
+    <source media="(prefers-color-scheme: dark)" srcset="./docs/extras/assets/qadence_arch.svg" width="75%">
+    <source media="(prefers-color-scheme: light)" srcset="./docs/extras/assets/qadence_arch.svg" width="75%">
+    <img alt="Qadence architecture" src="./docs/assets/qadence_arch.svg" width="75%">
+  </picture>
+<p align="center">
 
-* A [block-based system](docs/tutorials/getting_started.md) for composing _**complex digital-analog
+* A [block-based system](docs/content/block_system.md) for composing _**complex digital-analog
   programs**_ in a flexible and scalable manner, inspired by the Julia quantum SDK
   [Yao.jl](https://github.com/QuantumBFS/Yao.jl) and functional programming concepts.
 
-* A [simple interface](docs/digital_analog_qc/analog-basics.md) to work with _**interacting neutral-atom qubit systems**_
-  using [arbitrary registers topologies](docs/tutorials/register.md).
-
-* An intuitive [expression-based system](docs/tutorials/parameters.md) developed on top of the symbolic library [Sympy](https://www.sympy.org/en/index.html) to construct _**parametric quantum programs**_ easily.
-
-* [High-order generalized parameter shift rules](docs/advanced_tutorials/differentiability.md) for _**differentiating parametrized quantum operations**_.
+* An intuitive [expression-based system](docs/content/parameters.md) developed on top of the symbolic library [Sympy](https://www.sympy.org/en/index.html) to construct _**parametric quantum programs**_ easily.
 
 * Out-of-the-box _**automatic differentiability**_ of quantum programs with [PyTorch](https://pytorch.org/) integration.
 
-* _**Efficient execution**_ on a variety of different purpose backends: from state vector simulators to tensor network emulators and real devices.
+* [High-order generalized parameter shift rules](docs/tutorials/advanced_tutorials/differentiability.md) for _**differentiating parametrized quantum operations**_.
+
+* A [simple interface](docs/tutorials/digital_analog_qc/analog-basics.md) to work with _**interacting neutral-atom qubit systems**_
+  using [arbitrary registers topologies](docs/content/register.md).
 
 ## Installation guide
 
@@ -106,11 +108,10 @@ pip install qadence
 
 The default, pre-installed backend for Qadence is [PyQTorch](https://github.com/pasqal-io/pyqtorch), a differentiable state vector simulator for digital-analog simulation based on `PyTorch`. It is possible to install additional, `PyTorch` -based backends and the circuit visualization library using the following extras:
 
-* `pulser`: The [Pulser](https://github.com/pasqal-io/Pulser) backend for composing, simulating and executing pulse sequences for neutral-atom quantum devices.
-* `braket`: The [Braket](https://github.com/amazon-braket/amazon-braket-sdk-python) backend, an open source library that provides a framework for interacting with quantum computing hardware devices through Amazon Braket.
 * `visualization`: A visualization library to display quantum circuit diagrams.
 * `protocols`: A collection of [protocols](https://github.com/pasqal-io/qadence-protocols) for error mitigation in Qadence.
 * `libs`: A collection of [functionalities](https://github.com/pasqal-io/qadence-libs) for graph machine learning problems build on top of Qadence.
+* `pulser`: The [Pulser](https://github.com/pasqal-io/Pulser) backend for composing, simulating and executing pulse sequences for neutral-atom quantum devices (experimental).
 
 Qadence also supports a `JAX` engine which is currently supporting the [Horqrux](https://github.com/pasqal-io/horqrux) backend. `horqrux` is currently only available via the [low-level API](examples/backends/low_level/horqrux_backend.py).
 
@@ -118,7 +119,7 @@ Qadence also supports a `JAX` engine which is currently supporting the [Horqrux]
 To install individual extras, use the following syntax (**IMPORTANT** Make sure to use quotes):
 
 ```bash
-pip install "qadence[braket,pulser,visualization]"
+pip install "qadence[pulser,visualization]"
 ```
 
 To install all available extras, simply do:
@@ -144,10 +145,10 @@ conda install python-graphviz
 
 ## Contributing
 
-Before making a contribution, please review our [code of conduct](docs/CODE_OF_CONDUCT.md).
+Before making a contribution, please review our [code of conduct](docs/getting_started/CODE_OF_CONDUCT.md).
 
 - **Submitting Issues:** To submit bug reports or feature requests, please use our [issue tracker](https://github.com/pasqal-io/qadence/issues).
-- **Developing in qadence:** To learn more about how to develop within `qadence`, please refer to [contributing guidelines](docs/CONTRIBUTING.md).
+- **Developing in qadence:** To learn more about how to develop within `qadence`, please refer to [contributing guidelines](docs/getting_started/CONTRIBUTING.md).
 
 ### Setting up qadence in development mode
 

{qadence-1.5.0 → qadence-1.5.2}/README.md

@@ -1,16 +1,18 @@
-<picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/assets/logo/qadence_logo_white.svg">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/assets/logo/qadence_logo.svg">
-  <img alt="Qadence logo" src="./docs/assets/logo/qadence_logo.svg">
-</picture>
+<p align="center">
+  <picture>
+    <source media="(prefers-color-scheme: dark)" srcset="./docs/extras/assets/logo/qadence_logo_white.svg" width="75%">
+    <source media="(prefers-color-scheme: light)" srcset="./docs/extras/assets/logo/qadence_logo.svg" width="75%">
+    <img alt="Qadence logo" src="./docs/assets/logo/qadence_logo.svg" width="75%">
+  </picture>
+</p>
+
+**Qadence** is a Python package that provides a simple interface to build **digital-analog quantum
+programs** with tunable qubit interactions and arbitrary register topologies realizable on neutral atom devices.
 
 **For a high-level overview of Qadence features, [check out our white paper](https://arxiv.org/abs/2401.09915).**
 
 **For more detailed information, [check out the documentation](https://pasqal-io.github.io/qadence/latest/).**
 
-**Qadence** is a Python package that provides a simple interface to build _**digital-analog quantum
-programs**_ with tunable qubit interaction defined on _**arbitrary register topologies**_ realizable on neutral atom devices.
-
 [![Linting](https://github.com/pasqal-io/qadence/actions/workflows/lint.yml/badge.svg)](https://github.com/pasqal-io/qadence/actions/workflows/lint.yml)
 [![Tests](https://github.com/pasqal-io/qadence/actions/workflows/test_fast.yml/badge.svg)](https://github.com/pasqal-io/qadence/actions/workflows/test_fast.yml)
 [![Documentation](https://github.com/pasqal-io/qadence/actions/workflows/build_docs.yml/badge.svg)](https://pasqal-io.github.io/qadence/latest)
@@ -19,27 +21,26 @@ programs**_ with tunable qubit interaction defined on _**arbitrary register topo
 
 ## Feature highlights
 
-<picture>
-  <source media="(prefers-color-scheme: dark)" srcset="./docs/assets/qadence_arch.svg">
-  <source media="(prefers-color-scheme: light)" srcset="./docs/assets/qadence_arch.svg">
-  <img alt="Qadence architecture" src="./docs/assets/qadence_arch.svg">
-</picture>
-
+<p align="center">
+  <picture>
+    <source media="(prefers-color-scheme: dark)" srcset="./docs/extras/assets/qadence_arch.svg" width="75%">
+    <source media="(prefers-color-scheme: light)" srcset="./docs/extras/assets/qadence_arch.svg" width="75%">
+    <img alt="Qadence architecture" src="./docs/assets/qadence_arch.svg" width="75%">
+  </picture>
+<p align="center">
 
-* A [block-based system](docs/tutorials/getting_started.md) for composing _**complex digital-analog
+* A [block-based system](docs/content/block_system.md) for composing _**complex digital-analog
   programs**_ in a flexible and scalable manner, inspired by the Julia quantum SDK
   [Yao.jl](https://github.com/QuantumBFS/Yao.jl) and functional programming concepts.
 
-* A [simple interface](docs/digital_analog_qc/analog-basics.md) to work with _**interacting neutral-atom qubit systems**_
-  using [arbitrary registers topologies](docs/tutorials/register.md).
-
-* An intuitive [expression-based system](docs/tutorials/parameters.md) developed on top of the symbolic library [Sympy](https://www.sympy.org/en/index.html) to construct _**parametric quantum programs**_ easily.
-
-* [High-order generalized parameter shift rules](docs/advanced_tutorials/differentiability.md) for _**differentiating parametrized quantum operations**_.
+* An intuitive [expression-based system](docs/content/parameters.md) developed on top of the symbolic library [Sympy](https://www.sympy.org/en/index.html) to construct _**parametric quantum programs**_ easily.
 
 * Out-of-the-box _**automatic differentiability**_ of quantum programs with [PyTorch](https://pytorch.org/) integration.
 
-* _**Efficient execution**_ on a variety of different purpose backends: from state vector simulators to tensor network emulators and real devices.
+* [High-order generalized parameter shift rules](docs/tutorials/advanced_tutorials/differentiability.md) for _**differentiating parametrized quantum operations**_.
+
+* A [simple interface](docs/tutorials/digital_analog_qc/analog-basics.md) to work with _**interacting neutral-atom qubit systems**_
+  using [arbitrary registers topologies](docs/content/register.md).
 
 ## Installation guide
 
@@ -51,11 +52,10 @@ pip install qadence
 
 The default, pre-installed backend for Qadence is [PyQTorch](https://github.com/pasqal-io/pyqtorch), a differentiable state vector simulator for digital-analog simulation based on `PyTorch`. It is possible to install additional, `PyTorch` -based backends and the circuit visualization library using the following extras:
 
-* `pulser`: The [Pulser](https://github.com/pasqal-io/Pulser) backend for composing, simulating and executing pulse sequences for neutral-atom quantum devices.
-* `braket`: The [Braket](https://github.com/amazon-braket/amazon-braket-sdk-python) backend, an open source library that provides a framework for interacting with quantum computing hardware devices through Amazon Braket.
 * `visualization`: A visualization library to display quantum circuit diagrams.
 * `protocols`: A collection of [protocols](https://github.com/pasqal-io/qadence-protocols) for error mitigation in Qadence.
 * `libs`: A collection of [functionalities](https://github.com/pasqal-io/qadence-libs) for graph machine learning problems build on top of Qadence.
+* `pulser`: The [Pulser](https://github.com/pasqal-io/Pulser) backend for composing, simulating and executing pulse sequences for neutral-atom quantum devices (experimental).
 
 Qadence also supports a `JAX` engine which is currently supporting the [Horqrux](https://github.com/pasqal-io/horqrux) backend. `horqrux` is currently only available via the [low-level API](examples/backends/low_level/horqrux_backend.py).
 
@@ -63,7 +63,7 @@ Qadence also supports a `JAX` engine which is currently supporting the [Horqrux]
 To install individual extras, use the following syntax (**IMPORTANT** Make sure to use quotes):
 
 ```bash
-pip install "qadence[braket,pulser,visualization]"
+pip install "qadence[pulser,visualization]"
 ```
 
 To install all available extras, simply do:
@@ -89,10 +89,10 @@ conda install python-graphviz
 
 ## Contributing
 
-Before making a contribution, please review our [code of conduct](docs/CODE_OF_CONDUCT.md).
+Before making a contribution, please review our [code of conduct](docs/getting_started/CODE_OF_CONDUCT.md).
 
 - **Submitting Issues:** To submit bug reports or feature requests, please use our [issue tracker](https://github.com/pasqal-io/qadence/issues).
-- **Developing in qadence:** To learn more about how to develop within `qadence`, please refer to [contributing guidelines](docs/CONTRIBUTING.md).
+- **Developing in qadence:** To learn more about how to develop within `qadence`, please refer to [contributing guidelines](docs/getting_started/CONTRIBUTING.md).
 
 ### Setting up qadence in development mode
 

qadence-1.5.2/mkdocs.yml

@@ -0,0 +1,171 @@
+site_name: Qadence
+repo_url: "https://github.com/pasqal-io/qadence/"
+site_url: "https://pasqal-io.github.io/qadence/"
+repo_name: "qadence"
+
+nav:
+  - Home: index.md
+
+  - Getting started:
+    - Installation: getting_started/installation.md
+    - Development & Info:
+      - Contributing: getting_started/CONTRIBUTING.md
+      - Code of conduct: getting_started/CODE_OF_CONDUCT.md
+      - License: getting_started/LICENSE.md
+
+  - Contents:
+    - Block system: content/block_system.md
+    - Parametric programs: content/parameters.md
+    - Quantum models: content/quantummodels.md
+    - Quantum registers: content/register.md
+    - State initialization: content/state_init.md
+    - Arbitrary Hamiltonians: content/hamiltonians.md
+    - QML Constructors: content/qml_constructors.md
+    - Wavefunction overlaps: content/overlap.md
+    - Backends: content/backends.md
+    - Useful info:
+      - State conventions: content/state_conventions.md
+      - Serialization: content/serializ_and_prep.md
+
+  - Tutorials:
+    - tutorials/index.md
+    - Digital-analog quantum computing:
+      - tutorials/digital_analog_qc/index.md
+      - Basic operations on neutral-atoms: tutorials/digital_analog_qc/analog-basics.md
+      - Fitting a simple function: tutorials/digital_analog_qc/analog-qcl.md
+      - Restricted local addressability: tutorials/digital_analog_qc/semi-local-addressing.md
+      - Pulse-level programming with Pulser: tutorials/digital_analog_qc/pulser-basic.md
+      - Solve a QUBO problem: tutorials/digital_analog_qc/analog-qubo.md
+      - CNOT with interacting qubits: tutorials/digital_analog_qc/daqc-cnot.md
+
+    - Variational quantum algorithms:
+      - tutorials/qml/index.md
+      - Training tools: tutorials/qml/ml_tools.md
+      - Quantum circuit learning: tutorials/qml/qcl.md
+      - Solving MaxCut with QAOA: tutorials/qml/qaoa.md
+      - Solving a 1D ODE: tutorials/qml/dqc_1d.md
+
+    - Advanced Tutorials:
+      - tutorials/advanced_tutorials/index.md
+      - Quantum circuits differentiation: tutorials/advanced_tutorials/differentiability.md
+      - Custom quantum models: tutorials/advanced_tutorials/custom-models.md
+      - Projector blocks: tutorials/advanced_tutorials/projectors.md
+
+    - Realistic simulations:
+      - tutorials/realistic_sims/index.md
+      - Measurement protocols: tutorials/realistic_sims/measurements.md
+      - Simulated errors: tutorials/realistic_sims/noise.md
+      - Error mitigation: tutorials/realistic_sims/mitigation.md
+
+    - Development:
+      - Architecture and sharp bits: tutorials/development/architecture.md
+      - Drawing: tutorials/development/draw.md
+
+  - API:
+    - Block system: api/blocks.md
+    - Operations: api/operations.md
+    - Register: api/register.md
+    - QuantumCircuit: api/quantumcircuit.md
+    - Parameters: api/parameters.md
+    - State preparation: api/states.md
+    - Constructors: api/constructors.md
+    - Transpilation: api/transpile.md
+    - Execution: api/execution.md
+    - Quantum models: api/models.md
+    - QML tools: api/ml_tools.md
+    - Serialization: api/serialization.md
+    - Types: api/types.md
+    - Drawing: api/draw.md
+
+    - Backends:
+      - Abstract backend: api/backends/backend.md
+      - PyQTorch: api/backends/pyqtorch.md
+      - Amazon Braket: api/backends/braket.md
+      - Pulser: api/backends/pulser.md
+      - DifferentiableBackend: api/backends/differentiable.md
+
+edit_uri: edit/main/docs/
+
+theme:
+  name: material
+  custom_dir: docs/extras/overrides/
+  logo: extras/assets/logo/qadence_logo_small_white.svg
+  features:
+  - content.code.annotate
+  - navigation.tabs
+  - navigation.indexes
+  - navigation.sections
+  - content.code.copy
+  - content.code.annotate
+
+  palette:
+  - media: "(prefers-color-scheme: light)"
+    scheme: default
+    primary: custom
+    accent: custom
+    toggle:
+        icon: material/weather-sunny
+        name: Switch to dark mode
+  - media: "(prefers-color-scheme: dark)"
+    scheme: slate
+    primary: black
+    accent: custom
+    toggle:
+        icon: material/weather-night
+        name: Switch to light mode
+
+markdown_extensions:
+- admonition  # for notes
+- footnotes
+- pymdownx.arithmatex: # for mathjax
+    generic: true
+- pymdownx.highlight:
+    anchor_linenums: true
+- pymdownx.inlinehilite
+- pymdownx.snippets
+- pymdownx.details
+- pymdownx.superfences:
+    custom_fences:
+    - name: python
+      class: python
+      validator: "!!python/name:markdown_exec.validator"
+      format: "!!python/name:markdown_exec.formatter"
+
+plugins:
+- search
+- section-index
+- mkdocstrings:
+    default_handler: python
+    handlers:
+      python:
+        selection:
+          filters:
+            - "!^_"  # exlude all members starting with _
+            - "^__init__$"  # but always include __init__ modules and methods
+            - "^__new__$"  # but always include __init__ modules and methods
+        options:
+          show_root_toc_entry: false
+          heading_level: 3
+          merge_init_into_class: true
+          docstring_section_style: spacy
+
+- markdown-exec
+
+extra:
+ version:
+   provider: mike
+
+# To get nice tabs
+extra_css:
+- extras/css/mkdocstrings.css
+- extras/css/colors.css
+- extras/css/home.css
+
+# For mathjax
+extra_javascript:
+  - extras/javascripts/mathjax.js
+  - https://polyfill.io/v3/polyfill.min.js?features=es6
+  - https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js
+
+watch:
+- qadence

{qadence-1.5.0 → qadence-1.5.2}/pyproject.toml

@@ -17,10 +17,11 @@ authors = [
     { name = "Joao P. Moutinho", email = "joao.moutinho@pasqal.com"},
     { name = "Vytautas Abramavicius", email = "vytautas.abramavicius@pasqal.com" },
     { name = "Gergana Velikova", email = "gergana.velikova@pasqal.com" },
+    { name = "Eduardo Maschio", email = "eduardo.maschio@pasqal.com" },
 ]
 requires-python = ">=3.9,<3.13"
 license = {text = "Apache 2.0"}
-version = "1.5.0"
+version = "1.5.2"
 classifiers=[
     "License :: OSI Approved :: Apache Software License",
     "Programming Language :: Python",
@@ -44,6 +45,7 @@ dependencies = [
     "scipy",
     "pyqtorch==1.1.0",
     "matplotlib",
+    "Arpeggio==2.0.2",
 ]
 
 [tool.hatch.metadata]
@@ -71,7 +73,7 @@ protocols = ["qadence-protocols"]
 libs = ["qadence-libs"]
 all = [
   "pulser>=0.15.2",
-  "amazon-braket-sdk",
+  "amazon-braket-sdk==1.71.0",
   "graphviz",
   "protocols",
   "libs",

{qadence-1.5.0 → qadence-1.5.2}/qadence/backend.py

@@ -26,7 +26,6 @@ from qadence.mitigations import Mitigations
 from qadence.noise import Noise
 from qadence.parameters import stringify
 from qadence.types import ArrayLike, BackendName, DiffMode, Endianness, Engine, ParamDictType
-from qadence.utils import validate_values_and_state
 
 logger = get_logger(__file__)
 
@@ -259,29 +258,6 @@ class Backend(ABC):
         """
         raise NotImplementedError
 
-    @abstractmethod
-    def _run(
-        self,
-        circuit: ConvertedCircuit,
-        param_values: dict[str, ArrayLike] = {},
-        state: ArrayLike | None = None,
-        endianness: Endianness = Endianness.BIG,
-    ) -> ArrayLike:
-        """Run a circuit and return the resulting wave function.
-
-        Arguments:
-            circuit: A converted circuit as returned by `backend.circuit`.
-            param_values: _**Already embedded**_ parameters of the circuit. See
-                [`embedding`][qadence.blocks.embedding.embedding] for more info.
-            state: Initial state.
-            endianness: Endianness of the resulting wavefunction.
-
-        Returns:
-            A list of Counter objects where each key represents a bitstring
-            and its value the number of times it has been sampled from the given wave function.
-        """
-        raise NotImplementedError
-
     def run(
         self,
         circuit: ConvertedCircuit,
@@ -304,8 +280,7 @@ class Backend(ABC):
             A list of Counter objects where each key represents a bitstring
             and its value the number of times it has been sampled from the given wave function.
         """
-        validate_values_and_state(state, circuit.abstract.n_qubits, param_values)
-        return self._run(circuit, param_values, state, endianness, *args, **kwargs)
+        raise NotImplementedError
 
     @abstractmethod
     def run_dm(

{qadence-1.5.0 → qadence-1.5.2}/qadence/backends/braket/backend.py

@@ -88,7 +88,7 @@ class Backend(BackendInterface):
         ).squeeze(0)
         return ConvertedObservable(native=native, abstract=obs, original=obs)
 
-    def _run(
+    def run(
         self,
         circuit: ConvertedCircuit,
         param_values: dict[str, Tensor] = {},

{qadence-1.5.0 → qadence-1.5.2}/qadence/backends/horqrux/backend.py

@@ -66,7 +66,7 @@ class Backend(BackendInterface):
         hq_obs = convert_observable(block, n_qubits=n_qubits, config=self.config)
         return ConvertedObservable(native=hq_obs, abstract=block, original=observable)
 
-    def _run(
+    def run(
         self,
         circuit: ConvertedCircuit,
         param_values: ParamDictType = {},

{qadence-1.5.0 → qadence-1.5.2}/qadence/backends/pulser/backend.py

@@ -200,7 +200,7 @@ class Backend(BackendInterface):
 
         return circuit.native.build(**numpy_param_values)
 
-    def _run(
+    def run(
         self,
         circuit: ConvertedCircuit,
         param_values: dict[str, Tensor] = {},

{qadence-1.5.0 → qadence-1.5.2}/qadence/backends/pyqtorch/backend.py

@@ -80,7 +80,7 @@ class Backend(BackendInterface):
         (native,) = convert_observable(block, n_qubits=n_qubits, config=self.config)
         return ConvertedObservable(native=native, abstract=block, original=observable)
 
-    def _run(
+    def run(
         self,
         circuit: ConvertedCircuit,
         param_values: dict[str, Tensor] = {},

{qadence-1.5.0 → qadence-1.5.2}/qadence/ml_tools/models.py

@@ -289,11 +289,23 @@ class TransformedModule(torch.nn.Module):
     def to(self, *args: Any, **kwargs: Any) -> TransformedModule:
         try:
             self.model = self.model.to(*args, **kwargs)
-            self._input_scaling = self._input_scaling.to(*args, **kwargs)
-            self._input_shifting = self._input_shifting.to(*args, **kwargs)
-            self._output_scaling = self._output_scaling.to(*args, **kwargs)
-            self._output_shifting = self._output_shifting.to(*args, **kwargs)
-
+            if isinstance(self.model, QuantumModel):
+                device = self.model._circuit.native.device
+                dtype = (
+                    torch.float64
+                    if self.model._circuit.native.dtype == torch.cdouble
+                    else torch.float32
+                )
+
+                self._input_scaling = self._input_scaling.to(device=device, dtype=dtype)
+                self._input_shifting = self._input_shifting.to(device=device, dtype=dtype)
+                self._output_scaling = self._output_scaling.to(device=device, dtype=dtype)
+                self._output_shifting = self._output_shifting.to(device=device, dtype=dtype)
+            elif isinstance(self.model, torch.nn.Module):
+                self._input_scaling = self._input_scaling.to(*args, **kwargs)
+                self._input_shifting = self._input_shifting.to(*args, **kwargs)
+                self._output_scaling = self._output_scaling.to(*args, **kwargs)
+                self._output_shifting = self._output_shifting.to(*args, **kwargs)
             logger.debug(f"Moved {self} to {args}, {kwargs}.")
         except Exception as e:
             logger.warning(f"Unable to move {self} to {args}, {kwargs} due to {e}.")

{qadence-1.5.0 → qadence-1.5.2}/qadence/ml_tools/train_grad.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 from typing import Callable, Union
 
 from rich.progress import BarColumn, Progress, TaskProgressColumn, TextColumn, TimeRemainingColumn
+from torch import complex128, float32, float64
 from torch import device as torch_device
 from torch import dtype as torch_dtype
 from torch.nn import DataParallel, Module
@@ -110,17 +111,17 @@ def train(
     train_with_grad(model, data, optimizer, config, loss_fn=loss_fn)
     ```
     """
+    # load available checkpoint
+    init_iter = 0
+    if config.folder:
+        model, optimizer, init_iter = load_checkpoint(config.folder, model, optimizer)
+        logger.debug(f"Loaded model and optimizer from {config.folder}")
 
     # Move model to device before optimizer is loaded
     if isinstance(model, DataParallel):
         model = model.module.to(device=device, dtype=dtype)
     else:
         model = model.to(device=device, dtype=dtype)
-    # load available checkpoint
-    init_iter = 0
-    if config.folder:
-        model, optimizer, init_iter = load_checkpoint(config.folder, model, optimizer)
-        logger.debug(f"Loaded model and optimizer from {config.folder}")
     # initialize tensorboard
     writer = SummaryWriter(config.folder, purge_step=init_iter)
 
@@ -131,7 +132,9 @@ def train(
         TaskProgressColumn(),
         TimeRemainingColumn(elapsed_when_finished=True),
     )
-
+    data_dtype = None
+    if dtype:
+        data_dtype = float64 if dtype == complex128 else float32
     with progress:
         dl_iter = iter(dataloader) if dataloader is not None else None
 
@@ -143,7 +146,12 @@ def train(
                 # which do not have classical input data (e.g. chemistry)
                 if dataloader is None:
                     loss, metrics = optimize_step(
-                        model=model, optimizer=optimizer, loss_fn=loss_fn, xs=None, device=device
+                        model=model,
+                        optimizer=optimizer,
+                        loss_fn=loss_fn,
+                        xs=None,
+                        device=device,
+                        dtype=data_dtype,
                     )
                     loss = loss.item()
 
@@ -154,6 +162,7 @@ def train(
                         loss_fn=loss_fn,
                         xs=next(dl_iter),  # type: ignore[arg-type]
                         device=device,
+                        dtype=data_dtype,
                     )
 
                 else:

{qadence-1.5.0 → qadence-1.5.2}/qadence/models/quantum_model.py

@@ -342,9 +342,10 @@ class QuantumModel(nn.Module):
         return self.backend.assign_parameters(self._circuit, params)
 
     def to(self, *args: Any, **kwargs: Any) -> QuantumModel:
+        from pyqtorch import QuantumCircuit as PyQCircuit
+
         try:
-            if isinstance(self._circuit.native, torch.nn.Module):
-                # Backends which are not torch-based cannot be moved to 'device'
+            if isinstance(self._circuit.native, PyQCircuit):
                 self._circuit.native = self._circuit.native.to(*args, **kwargs)
                 if self._observable is not None:
                     if isinstance(self._observable, ConvertedObservable):
@@ -359,6 +360,8 @@ class QuantumModel(nn.Module):
                     else torch.float32,
                 )
                 logger.debug(f"Moved {self} to {args}, {kwargs}.")
+            else:
+                logger.debug("QuantumModel.to only supports pyqtorch.QuantumCircuits.")
         except Exception as e:
             logger.warning(f"Unable to move {self} to {args}, {kwargs} due to {e}.")
         return self