Add KPZ to TOC (#336)

* Update _index_included.yaml to include KPZ experiment * Improve code efficiency * Update notebook with new timings
quantumlib · Sep 9, 2023 · e48d8ff · e48d8ff
1 parent d021621
commit e48d8ff
Show file tree

Hide file tree

Showing 3 changed files with 75 additions and 43 deletions.
diff --git a/docs/_index_included.yaml b/docs/_index_included.yaml
@@ -98,3 +98,11 @@ included:
   - heading: Ground State
     description: Create a toric code ground state.
     path: /cirq/experiments/toric_code/toric_code_ground_state
+
+- heading: Kardar-Parisi-Zhang Physics
+  description: Probe the dynamics of transferred magnetization in the Heisenberg spin chain.
+  options:
+    - cards
+  items:
+  - heading: Experiment example
+    path: /cirq/experiments/kpz/kpz
diff --git a/docs/kpz/kpz.ipynb b/docs/kpz/kpz.ipynb
diff --git a/recirq/kpz/experiment.py b/recirq/kpz/experiment.py
@@ -551,9 +551,9 @@ def run_experiment_amplitudes(
             num_right_initial = np.sum(initial_bitstr)
             probs = (
                 np.abs(
-                    sampler.compute_amplitudes(
-                        cirq.drop_terminal_measurements(qc), all_states
-                    )
+                    sampler.simulate(
+                        cirq.drop_terminal_measurements(qc)
+                    ).final_state_vector
                 )
                 ** 2
             )