Merging snapshot 26/05/26 (#654)

Co-authored-by: Luke Helt <luke.helt@gmail.com>

Author: apchytr

.github/CHANGELOG.md

@@ -2,6 +2,20 @@
 
 ### Breaking changes
 
+* Deprecated `CircuitComponent.ansatz` returning `None`.
+
+* Deprecated passing an `Ansatz` into `__mul__`.
+
+* Updated `ComputationalGraph` to make use of `mm_einsum` sublist style calls.
+
+* Deprecated `mm_einsum` support for `ArrayLike` operands.
+
+* Removed `DISCRETIZATION_METHOD`; Wigner computation now uses only Clenshaw.
+
+* Bumped minimum pydantic to version 2.11 to support @cached_property directly.
+
+* Dropped support for Python 3.11.
+
 * Deprecated `Circuit` in the `lab` module.
 
 * Updated `lab` to use the new `AnsatzFactory` under the hood.
@@ -53,8 +67,46 @@ implementation `math.cython_lattice.beamsplitter`.
 * Changed the API of `Coherent`, `DisplacedSqueezed`, `Dgate` from real `x`, `y` parameters to a complex `alpha`.
 [(#617)](https://github.com/XanaduAI/MrMustard/pull/617)
 
+### New features
+
+* Added the new `Circuit` class.
+
+* Added Siegel-disk Riemannian optimization (Bergman metric) for complex symmetric matrices with operator norm < 1, via `Variable.siegel` and `Optimizer(siegel_lr=...)`.
+
+* Introduced the `ComputationalGraph`, a class to represent computations where nodes are components and edges are tensor contractions.
+
 ### Improvements
 
+* Added support for Python 3.14.
+
+* Added native jax version of the vanilla hermite_renormalized function.
+
+* `Optimizer` now catches `KeyboardInterrupt` during `minimize` and returns the current variable values instead of crashing. Added an opt-in `parameter_history` flag to `Optimizer.minimize` that stores intermediate variable values at each step in `Optimizer.parameter_history`.
+
+* Updated `mm_einsum` to support a sublist signature. As a result, contractions are no longer constrained to the number of supported characters.
+
+* Added `fock_config` and `representation_config` to `ComputationalGraph`.
+
+* Updated `ComputationalGraph` to do things in standard order.
+
+* `oscillator_eigenstate` now supports arbitrary batch dimensions.
+
+* Updated built-in `CircuitComponent`s to expose a `name` argument for users.
+
+* Added `fock_diagonals_1leftover` (in `mrmustard.mathlib.lattice.strategies.fock_diagonals`), a numerically stable and exact algorithm for computing conditional density matrices.
+
+* Added the `CircuitComponent.from_fock` classmethod.
+
+* Added `WIGNER_2D_RESOLUTION` and `WIGNER_BOUNDS` settings for the single-mode state repr Wigner plot.
+
+* Fixed `hermite_renormalized`: user-provided `stable` now overrides `STABLE_FOCK_CONVERSION`.
+
+* Added `visualize_2d_with_arrows` to `State`.
+
+* Made `fock_diagonals` numerically stable (in `mrmustard.mathlib.lattice.strategies.fock_diagonals`). It computes the multimode Fock diagonal / photon-number probabilities of a Gaussian operator.
+
+* Added `max_disp` as a keyword argument to `GaussianKet.random` and `GaussianDM.random`. When given, each mode is displaced by a random complex amplitude sampled uniformly from the disk of radius `max_disp`.
+
 * Added `tan`, `tanh`, `argmax`, `argmin`, `argsort`, and `mean` functions to the math backend.
 
 * Added `stellar_roots()` and `plot_stellar_roots()` methods to `Ket` for computing and visualizing the zeros of the Bargmann polynomial of single-mode states.
@@ -133,6 +185,10 @@ matrix is provided and a random interferometer is to be generated.
 * Added ability to raise when contracting two circuit components in different
 representations by setting `settings.DEFAULT_REPRESENTATION` to `None`.
 
+* Added the `Kgate` (Kerr) non-Gaussian single-mode unitary, diagonal in the Fock basis
+with entries :math:`e^{i\kappa n*(n-1)}`.
+[(#653)](https://github.com/XanaduAI/MrMustard/pull/653)
+
 * Added support for Python 3.13.
 [(#640)](https://github.com/XanaduAI/MrMustard/pull/640)
 
@@ -151,14 +207,18 @@ representations by setting `settings.DEFAULT_REPRESENTATION` to `None`.
 * Added a ``rich`` based repr to ``ParameterSet``.
 [(#616)](https://github.com/XanaduAI/MrMustard/pull/616)
 
-* Added the `Kgate` (Kerr) non-Gaussian single-mode unitary, diagonal in the Fock basis
-with entries :math:`e^{i\kappa n*(n-1)}`.
-[(#653)](https://github.com/XanaduAI/MrMustard/pull/653)
-
 ### Bug fixes
 
-* Fixed editable installs failing with a `FileNotFoundError` on `ninja` when re-running `uv run` (build isolation is now disabled for `mrmustard` so the rebuild hook uses the environment's stable `ninja`/`meson`).
-[(#655)](https://github.com/XanaduAI/MrMustard/pull/655)
+* Fixed a bug in `ComputationalGraph` where generic `CircuitComponent` subclasses were populating the `ansatz_dict`.
+
+* Tests will now all pass on GPU when selecting the JAX backend.
+
+* Moved `jax.config` logic from `mrmustard/math/backend_jax.py` into `mrmustard/__init__.py` to ensure settings are updated even when the backend is
+  not set to `jax`.
+
+* Added an absolute tolerance of `1e-20` to the check that a determinant in `complex_gaussian_integral_2_jitted` is close to `0` to mitigate underflows.
+
+* Fixed a bug in `mm_einsum` where the user-supplied path would only work for the Bargmann contraction. Now it works all the way through.
 
 * Fixed a bug in `CircuitComponent.quadrature` where einsum fock indices collided with mode indices for states on non-zero modes (e.g. a single-mode DM on mode 1 after tracing), forcing the number of quadrature points to equal the Fock cutoff and silently giving wrong results.
 
@@ -173,6 +233,9 @@ with entries :math:`e^{i\kappa n*(n-1)}`.
 
 * Fixed a bug that prevented `Number` from being JIT compiled in the JAX backend.
 
+* Fixed editable installs failing with a `FileNotFoundError` on `ninja` when re-running `uv run` (build isolation is now disabled for `mrmustard` so the rebuild hook uses the environment's stable `ninja`/`meson`).
+[(#655)](https://github.com/XanaduAI/MrMustard/pull/655)
+
 * Fixed bug in Gaussian integrals where small/large c values were not multiplying correctly.
 [(#641)](https://github.com/XanaduAI/MrMustard/pull/641)
 

.github/workflows/pre_commit.yml

@@ -12,7 +12,7 @@ jobs:
     - name: Install uv
       uses: astral-sh/setup-uv@v5
       with:
-        python-version: "3.13"
+        python-version: "3.14"
 
     - name: Run pre-commit
       run: uv run --frozen --only-dev pre-commit run --all-files

.github/workflows/tests_docs.yml

@@ -16,8 +16,6 @@ jobs:
     concurrency:
       group: ${{ github.workflow }}-${{ github.ref }}
       cancel-in-progress: true
-    env:
-      HYPOTHESIS_PROFILE: ci
 
     steps:
       - name: Checkout repo
@@ -26,7 +24,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@v5
         with:
-          python-version: "3.13"
+          python-version: "3.14"
 
       - name: Install system dependencies
         run: sudo apt-get update && sudo apt-get install -y graphviz

.github/workflows/tests_jax.yml

@@ -18,10 +18,9 @@ jobs:
       id-token: write # This is required for requesting the JWT for S3 access
     strategy:
       matrix:
-        python-version: ['3.11', '3.13']
+        python-version: ['3.12', '3.14']
     env:
-      HYPOTHESIS_PROFILE: ci
-      UPLOAD_TIMINGS: ${{ github.event_name != 'pull_request' && matrix.python-version == '3.13' && 'true' || 'false' }}
+      UPLOAD_TIMINGS: ${{ github.event_name != 'pull_request' && matrix.python-version == '3.14' && 'true' || 'false' }}
 
     steps:
       - name: Checkout repo

.github/workflows/tests_numpy.yml

@@ -18,10 +18,9 @@ jobs:
       id-token: write # This is required for requesting the JWT for S3 access
     strategy:
       matrix:
-        python-version: ['3.11', '3.13']
+        python-version: ['3.12', '3.14']
     env:
-      HYPOTHESIS_PROFILE: ci
-      UPLOAD_TIMINGS: ${{ github.event_name != 'pull_request' && matrix.python-version == '3.13' && 'true' || 'false' }}
+      UPLOAD_TIMINGS: ${{ github.event_name != 'pull_request' && matrix.python-version == '3.14' && 'true' || 'false' }}
 
     steps:
       - name: Checkout repo

.github/workflows/upload.yml

@@ -6,8 +6,6 @@ on:
 jobs:
   upload:
     runs-on: ubuntu-latest
-    env:
-      HYPOTHESIS_PROFILE: ci
 
     steps:
       - name: Checkout repo
@@ -16,7 +14,7 @@ jobs:
       - name: Install uv
         uses: astral-sh/setup-uv@v5
         with:
-          python-version: "3.13"
+          python-version: "3.14"
 
       - name: Build and install Mr Mustard
         run: |

.pre-commit-config.yaml

@@ -3,7 +3,6 @@ repos:
     rev: v0.18.1
     hooks:
     -   id: cython-lint
-        args: [--max-line-length=100]
     -   id: double-quote-cython-strings
 -   repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v2.3.0
@@ -12,8 +11,12 @@ repos:
     - id: check-merge-conflict
       args: [--assume-in-merge]
       exclude: \.rst$
+-   repo: https://github.com/astral-sh/uv-pre-commit
+    rev: 0.8.11
+    hooks:
+      - id: uv-lock
 - repo: https://github.com/astral-sh/ruff-pre-commit
-  rev: v0.12.0
+  rev: v0.15.7
   hooks:
     - id: ruff-check
       args: [ --fix, --exit-non-zero-on-fix]

.readthedocs.yml

@@ -5,7 +5,7 @@ version: 2
 build:
   os: ubuntu-22.04
   tools:
-    python: "3.11"
+    python: "3.14"
   apt_packages:
     - graphviz
   jobs:

doc/code/lab.rst

@@ -8,8 +8,10 @@ The `lab` module
 .. toctree::
     :maxdepth: 1
 
+    lab/circuit
     lab/circuit_components
     lab/circuit_components_utils
+    lab/computational_graph
     lab/samplers
     lab/states
     lab/transformations

doc/code/lab/circuit.rst

@@ -0,0 +1,16 @@
+.. currentmodule:: mrmustard.lab.circuit
+
+Circuit
+=======
+
+.. autosummary::
+    :nosignatures:
+
+    Circuit
+
+
+Circuit
+-------
+
+.. autoclass:: mrmustard.lab.circuit::Circuit
+    :members: