{
"cells": [
{
"cell_type": "markdown",
"id": "f67c4af7",
"metadata": {},
"source": [
"# Discriminated Single Shot Readout\n",
"The notebook will show how to run a readout calibration experiment and fit a discriminator with a linear discriminant analysis.\n",
"This experiment is sometimes called multi-state discrimination."
]
},
{
"cell_type": "code",
"execution_count": 1,
"id": "958b49c9",
"metadata": {
"execution": {
"iopub.execute_input": "2024-10-17T13:11:46.366101Z",
"iopub.status.busy": "2024-10-17T13:11:46.365811Z",
"iopub.status.idle": "2024-10-17T13:11:48.061037Z",
"shell.execute_reply": "2024-10-17T13:11:48.060180Z"
}
},
"outputs": [],
"source": [
"from functools import partial\n",
"from typing import Literal\n",
"\n",
"import matplotlib.pyplot as plt\n",
"import numpy as np\n",
"from qcodes.parameters import ManualParameter\n",
"from sklearn.metrics import ConfusionMatrixDisplay\n",
"\n",
"import quantify_core.data.handling as dh\n",
"from quantify_core.analysis.readout_calibration_analysis import ReadoutCalibrationAnalysis\n",
"from quantify_scheduler import Schedule\n",
"from quantify_scheduler.enums import BinMode\n",
"from quantify_scheduler.gettables import ScheduleGettable\n",
"from quantify_scheduler.operations.gate_library import Measure, Reset, Rxy"
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "65d9a524",
"metadata": {
"execution": {
"iopub.execute_input": "2024-10-17T13:11:48.064063Z",
"iopub.status.busy": "2024-10-17T13:11:48.063317Z",
"iopub.status.idle": "2024-10-17T13:11:48.071662Z",
"shell.execute_reply": "2024-10-17T13:11:48.071035Z"
}
},
"outputs": [],
"source": [
"import json\n",
"\n",
"import rich # noqa:F401\n",
"\n",
"import quantify_core.data.handling as dh\n",
"from quantify_scheduler.device_under_test.quantum_device import QuantumDevice\n",
"\n",
"from utils import initialize_hardware, run # noqa:F401"
]
},
{
"cell_type": "markdown",
"id": "6d97e2e4",
"metadata": {},
"source": [
"## Setup\n",
"In this section we configure the hardware configuration which specifies the connectivity of our system.\n",
"\n",
"The experiments of this tutorial are meant to be executed with a Qblox Cluster controlling a transmon system.\n",
"The experiments can also be executed using a dummy Qblox device that is created via an instance of the `Cluster` class, and is initialized with a dummy configuration.\n",
"When using a dummy device, the analysis will not work because the experiments will return `np.nan` values.\n",
"\n",
"### Configuration file\n",
"\n",
"This is a template hardware configuration file for a 2-qubit system with a flux-control line which can be used to tune the qubit frequency. We will only work with qubit 0.\n",
"\n",
"The hardware connectivity is as follows, by cluster slot:\n",
"- **QCM** (Slot 2)\n",
" - $\\text{O}^{1}$: Flux line for `q0`.\n",
" - $\\text{O}^{2}$: Flux line for `q1`.\n",
"- **QCM-RF** (Slot 6)\n",
" - $\\text{O}^{1}$: Drive line for `q0` using fixed 80 MHz IF.\n",
" - $\\text{O}^{2}$: Drive line for `q1` using fixed 80 MHz IF.\n",
"- **QRM-RF** (Slot 8)\n",
" - $\\text{O}^{1}$ and $\\text{I}^{1}$: Shared readout line for `q0`/`q1` using a fixed LO set at 7.5 GHz.\n",
"\n",
"Note that in the hardware configuration below the mixers are uncorrected, but for high fidelity experiments this should also be done for all the modules."
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "cd363199",
"metadata": {
"execution": {
"iopub.execute_input": "2024-10-17T13:11:48.073648Z",
"iopub.status.busy": "2024-10-17T13:11:48.073479Z",
"iopub.status.idle": "2024-10-17T13:11:48.077903Z",
"shell.execute_reply": "2024-10-17T13:11:48.077187Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Data will be saved in:\n",
"/root/quantify-data\n"
]
}
],
"source": [
"with open(\"configs/tuning_transmon_coupled_pair_hardware_config.json\") as hw_cfg_json_file:\n",
" hardware_cfg = json.load(hw_cfg_json_file)\n",
"\n",
"# Enter your own dataset directory here!\n",
"dh.set_datadir(dh.default_datadir())"
]
},
{
"cell_type": "markdown",
"id": "d85bf350",
"metadata": {},
"source": [
"### Quantum device settings\n",
"Here we initialize our `QuantumDevice` and our qubit parameters, checkout this [tutorial](https://quantify-os.org/docs/quantify-scheduler/tutorials/Operations%20and%20Qubits.html) for further details.\n",
"\n",
"In short, a `QuantumDevice` contains device elements where we save our found parameters. Here we are loading a template for 2 qubits, but we will only use qubit 0."
]
},
{
"cell_type": "code",
"execution_count": 4,
"id": "0e83223d",
"metadata": {
"execution": {
"iopub.execute_input": "2024-10-17T13:11:48.080383Z",
"iopub.status.busy": "2024-10-17T13:11:48.080214Z",
"iopub.status.idle": "2024-10-17T13:11:48.612023Z",
"shell.execute_reply": "2024-10-17T13:11:48.611264Z"
}
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"/usr/local/lib/python3.9/site-packages/quantify_scheduler/backends/types/qblox.py:1220: ValidationWarning: Setting `auto_lo_cal=on_lo_interm_freq_change` will overwrite settings `dc_offset_i=0.0` and `dc_offset_q=0.0`. To suppress this warning, do not set either `dc_offset_i` or `dc_offset_q` for this port-clock.\n",
" warnings.warn(\n",
"/usr/local/lib/python3.9/site-packages/quantify_scheduler/backends/types/qblox.py:1235: ValidationWarning: Setting `auto_sideband_cal=on_interm_freq_change` will overwrite settings `amp_ratio=1.0` and `phase_error=0.0`. To suppress this warning, do not set either `amp_ratio` or `phase_error` for this port-clock.\n",
" warnings.warn(\n"
]
}
],
"source": [
"quantum_device = QuantumDevice.from_json_file(\"devices/transmon_device_2q.json\")\n",
"qubit = quantum_device.get_element(\"q0\")\n",
"quantum_device.hardware_config(hardware_cfg)\n",
"meas_ctrl, _, cluster = initialize_hardware(quantum_device, ip=None)"
]
},
{
"cell_type": "markdown",
"id": "8e0b6a43",
"metadata": {
"lines_to_next_cell": 0
},
"source": [
"## Schedule definition"
]
},
{
"cell_type": "code",
"execution_count": 5,
"id": "fc7f6c65",
"metadata": {
"execution": {
"iopub.execute_input": "2024-10-17T13:11:48.615145Z",
"iopub.status.busy": "2024-10-17T13:11:48.614955Z",
"iopub.status.idle": "2024-10-17T13:11:48.620521Z",
"shell.execute_reply": "2024-10-17T13:11:48.619785Z"
}
},
"outputs": [],
"source": [
"def readout_calibration_sched(\n",
" qubit: str,\n",
" prepared_states: list[int],\n",
" repetitions: int = 1,\n",
" acq_protocol: Literal[\n",
" \"SSBIntegrationComplex\", \"ThresholdedAcquisition\"\n",
" ] = \"SSBIntegrationComplex\",\n",
") -> Schedule:\n",
" \"\"\"\n",
" Make a schedule for readout calibration.\n",
"\n",
" Parameters\n",
" ----------\n",
" qubit\n",
" The name of the qubit e.g., :code:`\"q0\"` to perform the experiment on.\n",
" prepared_states\n",
" A list of integers indicating which state to prepare the qubit in before measuring.\n",
" The ground state corresponds to 0 and the first-excited state to 1.\n",
" repetitions\n",
" The number of times the schedule will be repeated. Fixed to 1 for this schedule.\n",
" acq_protocol\n",
" The acquisition protocol used for the readout calibration. By default\n",
" \"SSBIntegrationComplex\", but \"ThresholdedAcquisition\" can be\n",
" used for verifying thresholded acquisition parameters.\n",
"\n",
" Returns\n",
" -------\n",
" :\n",
" An experiment schedule.\n",
"\n",
" Raises\n",
" ------\n",
" NotImplementedError\n",
" If the prepared state is > 1.\n",
"\n",
" \"\"\"\n",
" schedule = Schedule(f\"Readout calibration {qubit}\", repetitions=1)\n",
"\n",
" for i, prep_state in enumerate(prepared_states):\n",
" schedule.add(Reset(qubit), label=f\"Reset {i}\")\n",
" if prep_state == 0:\n",
" pass\n",
" elif prep_state == 1:\n",
" schedule.add(Rxy(qubit=qubit, theta=180, phi=0))\n",
" else:\n",
" raise NotImplementedError(\n",
" \"Preparing the qubit in the higher excited states is not supported yet.\"\n",
" )\n",
" schedule.add(\n",
" Measure(qubit, acq_index=i, bin_mode=BinMode.APPEND, acq_protocol=acq_protocol),\n",
" label=f\"Measurement {i}\",\n",
" )\n",
" return schedule"
]
},
{
"cell_type": "markdown",
"id": "c4b99379",
"metadata": {
"lines_to_next_cell": 0
},
"source": [
"## SSRO with single side band (SSB) integration"
]
},
{
"cell_type": "code",
"execution_count": 6,
"id": "e91f547e",
"metadata": {
"execution": {
"iopub.execute_input": "2024-10-17T13:11:48.623213Z",
"iopub.status.busy": "2024-10-17T13:11:48.623032Z",
"iopub.status.idle": "2024-10-17T13:11:48.627291Z",
"shell.execute_reply": "2024-10-17T13:11:48.626558Z"
},
"lines_to_next_cell": 0
},
"outputs": [],
"source": [
"states = ManualParameter(name=\"states\", unit=\"\", label=\"Prepared state\")\n",
"states.batch_size = 400\n",
"states.batched = True\n",
"\n",
"readout_calibration_sched_kwargs = dict(\n",
" qubit=qubit.name, prepared_states=states, acq_protocol=\"SSBIntegrationComplex\"\n",
")\n",
"\n",
"# set gettable\n",
"ssro_gettable = ScheduleGettable(\n",
" quantum_device,\n",
" schedule_function=readout_calibration_sched,\n",
" schedule_kwargs=readout_calibration_sched_kwargs,\n",
" real_imag=True,\n",
" batched=True,\n",
")\n",
"\n",
"# set measurement control\n",
"meas_ctrl.gettables(ssro_gettable)"
]
},
{
"cell_type": "code",
"execution_count": 7,
"id": "f824de9c",
"metadata": {
"execution": {
"iopub.execute_input": "2024-10-17T13:11:48.629196Z",
"iopub.status.busy": "2024-10-17T13:11:48.629027Z",
"iopub.status.idle": "2024-10-17T13:11:48.781659Z",
"shell.execute_reply": "2024-10-17T13:11:48.780918Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Starting batched measurement...\n",
"Iterative settable(s) [outer loop(s)]:\n",
"\t --- (None) --- \n",
"Batched settable(s):\n",
"\t states \n",
"Batch size limit: 400\n",
"\n"
]
},
{
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "ca0b4d31076b4f659018f1311d77f7cd",
"version_major": 2,
"version_minor": 0
},
"text/plain": [
"Completed: 0%| [ elapsed time: 00:00 | time left: ? ] it"
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"text/html": [
"