examples/fluxonium_multipath_coupling/fluxonium_mpcoupling_6q_cnot.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Fluxonium with multi-path coupling(Simultaneous CNOT gate)\n",
    "\n",
    "This notebook is similar but not as detail as `examples/fluxonium_multipath_coupling/fluxonium_mpcoupling_6q_x.ipynb`. Please check that notebook for more info.\n",
    "\n",
    "This example is based on Nguyen, L. B. et al. Blueprint for a High-Performance Fluxonium Quantum Processor. PRX Quantum 3, 037001 (2022).\n",
    "We simulate a 6 fluxonium qubit system from an underlying periodic lattice.\n",
    "Idling hamiltonian of the system is\n",
    "$$ H(0) = \\sum_i H_{\\mathrm{f},i}  + \\sum_{\\langle i,j \\rangle } H_{ij}. $$\n",
    "Hamiltonian of single fluxonium is\n",
    "$$ H_{\\mathrm{f},i} = 4E_{\\mathrm{C},i} n_i^2 + \\frac{1}{2} E_{\\mathrm{L},i} (\\varphi_i +\\varphi_{\\mathrm{ext},i})^2\n",
    "    -E_{\\mathrm{J},i}\\cos \\left( \\varphi_i \\right).$$\n",
    "The coupling terms have the form\n",
    "$$H_{ij} = J_{\\mathrm{C}} n_i n_j - J_{\\mathrm{L}} \\varphi_i \\varphi_j.$$\n",
    "The couplings are chosen in a way such that the idling $ZZ$-crosstalk is almost zero."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import jax\n",
    "import jax.numpy as jnp\n",
    "import haiku as hk\n",
    "\n",
    "from supergrad.helper import Evolve\n",
    "from supergrad.utils import tensor, permute, compute_fidelity_with_1q_rotation_axis\n",
    "from supergrad.utils.optimize import scipy_minimize, adam_opt\n",
    "from supergrad.utils.gates import cnot\n",
    "from supergrad.utils.utility import tree_print\n",
    "\n",
    "from supergrad.scgraph.graph_mpc_fluxonium_5x5_periodic import CNOTGatePeriodicGraphOpt\n",
    "\n",
    "truncated_dim = 3  # how many levels we keep after diagonalization\n",
    "add_random = True  # whether to add random deviations to fluxonium parameters ec,ej,el\n",
    "share_params = True  # whether we treat marked parameters as shared when computing gradients\n",
    "unify_coupling = True  # set all couplings to be equal\n",
    "\n",
    "compensation_option = 'arbit_single'  # allow arbitary single-qubit gate compensation\n",
    "# This is useful for checking how many errors are fundamentally multi-qubit\n",
    "\n",
    "# This specify one order of the qubits such that the simultaneous gates are [cnot()] * 3\n",
    "unitary_order = ['q02', 'q03', 'q12', 'q13', 'q22', 'q23']\n",
    "\n",
    "# instance the quantum processor graph, and choose a subset for time evolution\n",
    "gate_graph = CNOTGatePeriodicGraphOpt(seed=1)\n",
    "qubit_subset = gate_graph.subgraph(unitary_order)\n",
    "opt = 'adam'\n",
    "\n",
    "evo = Evolve(qubit_subset, truncated_dim, add_random, share_params, unify_coupling, compensation_option)\n",
    "\n",
    "# Compute the order of qubits in qubit_subset related to unitary_order\n",
    "qubit_order = [unitary_order.index(key) for key in qubit_subset.sorted_nodes]\n",
    "# Compute the corresponding target unitary based on the above order\n",
    "target_unitary = permute(tensor(*([cnot()] * 3)), [2] * len(qubit_subset.nodes), qubit_order)\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Let us print the parameters stored in the evo instance. These parameters are parsed from gate_graph. For how we construct this graph, please check `examples/fluxonium_multipath_coupling/graph_5x5_periodic.py` and the instructions inside."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{ 'capacitive_coupling_all_unify': {'strength': 0.07225663},\n",
      "  'inductive_coupling_all_unify': {'strength': -0.01256637},\n",
      "  'q02': {'ec': 6.3532393823, 'ej': 25.1103087675, 'el': 6.344787656},\n",
      "  'q02_pulse_rampcos': { 'amp': 0.18128846,\n",
      "                         'omega_d': 2.58934559,\n",
      "                         'phase': -0.24290228,\n",
      "                         't_plateau': 69.93608145,\n",
      "                         't_ramp': 29.92806488},\n",
      "  'q03': {'ec': 6.2550610547000005, 'ej': 25.1411401038, 'el': 5.000964059},\n",
      "  'q12': {'ec': 6.2233456874, 'ej': 25.1512697752, 'el': 5.6012793457},\n",
      "  'q12_pulse_rampcos': { 'amp': 0.17872194,\n",
      "                         'omega_d': 4.19989714,\n",
      "                         'phase': -0.01543561,\n",
      "                         't_plateau': 69.95327862,\n",
      "                         't_ramp': 29.94562879},\n",
      "  'q13': {'ec': 6.2867872105, 'ej': 25.1316550509, 'el': 6.9144529962},\n",
      "  'q22': { 'ec': 6.282846661600001,\n",
      "           'ej': 25.1329965509,\n",
      "           'el': 7.539286440900001},\n",
      "  'q22_pulse_rampcos': { 'amp': 0.23370657,\n",
      "                         'omega_d': 3.65018191,\n",
      "                         'phase': -0.50006247,\n",
      "                         't_plateau': 69.96302405,\n",
      "                         't_ramp': 29.98769304},\n",
      "  'single_q_compensation': { 'post_comp_q02': [ -0.01528741,\n",
      "                                                0.04418319,\n",
      "                                                -0.374539],\n",
      "                             'post_comp_q03': [ -1.00629771,\n",
      "                                                0.16411027,\n",
      "                                                -0.00797514],\n",
      "                             'post_comp_q12': [ 0.00200104,\n",
      "                                                0.58679953,\n",
      "                                                -1.74804011],\n",
      "                             'post_comp_q13': [ -0.11690826,\n",
      "                                                1.09973631,\n",
      "                                                -1.0941268],\n",
      "                             'post_comp_q22': [ -0.00142308406,\n",
      "                                                -1.59853235,\n",
      "                                                -0.188352743],\n",
      "                             'post_comp_q23': [ 1.65404972,\n",
      "                                                -0.67251125,\n",
      "                                                -0.04376101],\n",
      "                             'pre_comp_q02': [ 0.05671005,\n",
      "                                               0.02682845,\n",
      "                                               0.0395099],\n",
      "                             'pre_comp_q03': [ -1.46070525,\n",
      "                                               0.66029355,\n",
      "                                               0.06603344],\n",
      "                             'pre_comp_q12': [ 0.00185588,\n",
      "                                               1.00099539,\n",
      "                                               -0.22439037],\n",
      "                             'pre_comp_q13': [ -1.02837338,\n",
      "                                               -1.10918103,\n",
      "                                               -0.80985523],\n",
      "                             'pre_comp_q22': [ -0.000190040736,\n",
      "                                               -1.34752499,\n",
      "                                               0.315152817],\n",
      "                             'pre_comp_q23': [ 0.36516325,\n",
      "                                               -0.38763126,\n",
      "                                               -0.41575102]}}\n"
     ]
    }
   ],
   "source": [
    "params = evo.all_params\n",
    "tree_print(params)\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "For optimization, let us use an object function based on the average gate fidelity with leakage. The formula is from Physical Review A 87, 022309 (2013).\n",
    "$$\n",
    "\tC = 1 - \\frac{1}{D(D + 1)}[Tr(PU^{\\dagger}_{\\text{sim}}P U_{\\text{sim}}P)+|Tr(PU^{\\dagger}_{\\text{sim}} P U_{\\text{target}})|^2],   \n",
    "$$\n",
    "where $P$ is the projector into the computational subspace $S$, and $D=2^n$ is the dimension of $S$."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "def infidelity(params, target_unitary):\n",
    "    params = hk.data_structures.merge(evo.all_params, params)\n",
    "    # Compute the time evolution unitary in the eigenbasis.\n",
    "    sim_u = evo.eigen_basis(params)\n",
    "    # calculate fidelity\n",
    "    fidelity_vz, _ = compute_fidelity_with_1q_rotation_axis(target_unitary,\n",
    "                                                            sim_u,\n",
    "                                                            opt_method=None)\n",
    "\n",
    "    return jnp.log10(1 - fidelity_vz)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "The infidelity of simultaneous X gates is  0.0006533218397565268\n"
     ]
    }
   ],
   "source": [
    "vg = jax.value_and_grad(infidelity)(params, target_unitary)\n",
    "print('The infidelity of simultaneous X gates is ', 10**vg[0])\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "The gradient of log10(infidelity):\n",
      "{ 'capacitive_coupling_all_unify': {'strength': -0.30678138960742607},\n",
      "  'inductive_coupling_all_unify': {'strength': -158.6583899707822},\n",
      "  'q02': { 'ec': 0.4434989549839017,\n",
      "           'ej': 0.12670299682426123,\n",
      "           'el': -0.1462045689537277},\n",
      "  'q02_pulse_rampcos': { 'amp': -2.2086894895044833,\n",
      "                         'omega_d': 5.41994775487869,\n",
      "                         'phase': 0.181825493321193,\n",
      "                         't_plateau': 0.00035738436408789973,\n",
      "                         't_ramp': 0.0007715123434355182},\n",
      "  'q03': { 'ec': -13.55767252401893,\n",
      "           'ej': 4.085588336631121,\n",
      "           'el': -12.431915770031264},\n",
      "  'q12': { 'ec': -6.45063715091898,\n",
      "           'ej': 2.017487089770461,\n",
      "           'el': -5.852363732422589},\n",
      "  'q12_pulse_rampcos': { 'amp': -0.03347321995285995,\n",
      "                         'omega_d': 4.2612528500534435,\n",
      "                         'phase': 0.11020943732382167,\n",
      "                         't_plateau': -0.004077695689920287,\n",
      "                         't_ramp': -0.007391031717304665},\n",
      "  'q13': { 'ec': 6.258583335054281,\n",
      "           'ej': -2.079992497590237,\n",
      "           'el': 5.471496887867498},\n",
      "  'q22': { 'ec': -3.6920381654549024,\n",
      "           'ej': 1.3852793785905508,\n",
      "           'el': -3.384415464266684},\n",
      "  'q22_pulse_rampcos': { 'amp': -1.2754302293244881,\n",
      "                         'omega_d': 1.9926449213562933,\n",
      "                         'phase': 0.0508842529563353,\n",
      "                         't_plateau': -0.07277041474880228,\n",
      "                         't_ramp': -0.14474397468578978},\n",
      "  'single_q_compensation': { 'post_comp_q02': [ -0.04305220859420708,\n",
      "                                                -0.10800249480161028,\n",
      "                                                -0.0004724756383009418],\n",
      "                             'post_comp_q03': [ -0.04024295101614683,\n",
      "                                                0.09250106586461015,\n",
      "                                                0.047733427649442356],\n",
      "                             'post_comp_q12': [ 0.02077283179089251,\n",
      "                                                -0.11103942442389325,\n",
      "                                                -0.0007926361334363893],\n",
      "                             'post_comp_q13': [ -0.05254880182152988,\n",
      "                                                0.13287177554479968,\n",
      "                                                -0.03548127978747484],\n",
      "                             'post_comp_q22': [ 0.025936382917775225,\n",
      "                                                -0.04342969434181754,\n",
      "                                                0.00013829329146460175],\n",
      "                             'post_comp_q23': [ -0.07141240300111917,\n",
      "                                                -0.08701813457128083,\n",
      "                                                -0.25254766933877093],\n",
      "                             'pre_comp_q02': [ 0.06123035103842867,\n",
      "                                               0.006268409232411898,\n",
      "                                               0.0002780178812252485],\n",
      "                             'pre_comp_q03': [ -0.1407932441540325,\n",
      "                                               0.16294339976798256,\n",
      "                                               -0.3949979608484213],\n",
      "                             'pre_comp_q12': [ 0.08403426229383482,\n",
      "                                               -0.11317221317483472,\n",
      "                                               -0.0013775863529980647],\n",
      "                             'pre_comp_q13': [ 0.06383886744009715,\n",
      "                                               0.01563041534518561,\n",
      "                                               -0.03848473592506707],\n",
      "                             'pre_comp_q22': [ 0.02596379373312538,\n",
      "                                               -0.04318281539184454,\n",
      "                                               0.00010639813376038231],\n",
      "                             'pre_comp_q23': [ 0.030388223891506527,\n",
      "                                               -0.011048125119941687,\n",
      "                                               0.06590365507868157]}}\n"
     ]
    }
   ],
   "source": [
    "print('The gradient of log10(infidelity):')\n",
    "tree_print(vg[1])\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Show the optimization procedure of control parameters"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'adam_lr': 1e-06, 'adam_lr_decay_rate': 1000, 'steps': 3, 'adam_b1': 0.9, 'adam_b2': 0.999}\n",
      "step: 0\n",
      "parameters:\n",
      "{ 'capacitive_coupling_all_unify': {'strength': 0.07225663},\n",
      "  'inductive_coupling_all_unify': {'strength': -0.01256637},\n",
      "  'q02': {'ec': 6.3532393823, 'ej': 25.1103087675, 'el': 6.344787656},\n",
      "  'q02_pulse_rampcos': { 'amp': 0.18128846,\n",
      "                         'omega_d': 2.58934559,\n",
      "                         'phase': -0.24290228,\n",
      "                         't_plateau': 69.93608145,\n",
      "                         't_ramp': 29.92806488},\n",
      "  'q03': {'ec': 6.2550610547000005, 'ej': 25.1411401038, 'el': 5.000964059},\n",
      "  'q12': {'ec': 6.2233456874, 'ej': 25.1512697752, 'el': 5.6012793457},\n",
      "  'q12_pulse_rampcos': { 'amp': 0.17872194,\n",
      "                         'omega_d': 4.19989714,\n",
      "                         'phase': -0.01543561,\n",
      "                         't_plateau': 69.95327862,\n",
      "                         't_ramp': 29.94562879},\n",
      "  'q13': {'ec': 6.2867872105, 'ej': 25.1316550509, 'el': 6.9144529962},\n",
      "  'q22': { 'ec': 6.282846661600001,\n",
      "           'ej': 25.1329965509,\n",
      "           'el': 7.539286440900001},\n",
      "  'q22_pulse_rampcos': { 'amp': 0.23370657,\n",
      "                         'omega_d': 3.65018191,\n",
      "                         'phase': -0.50006247,\n",
      "                         't_plateau': 69.96302405,\n",
      "                         't_ramp': 29.98769304},\n",
      "  'single_q_compensation': { 'post_comp_q02': [ -0.01528741,\n",
      "                                                0.04418319,\n",
      "                                                -0.374539],\n",
      "                             'post_comp_q03': [ -1.00629771,\n",
      "                                                0.16411027,\n",
      "                                                -0.00797514],\n",
      "                             'post_comp_q12': [ 0.00200104,\n",
      "                                                0.58679953,\n",
      "                                                -1.74804011],\n",
      "                             'post_comp_q13': [ -0.11690826,\n",
      "                                                1.09973631,\n",
      "                                                -1.0941268],\n",
      "                             'post_comp_q22': [ -0.00142308406,\n",
      "                                                -1.59853235,\n",
      "                                                -0.188352743],\n",
      "                             'post_comp_q23': [ 1.65404972,\n",
      "                                                -0.67251125,\n",
      "                                                -0.04376101],\n",
      "                             'pre_comp_q02': [ 0.05671005,\n",
      "                                               0.02682845,\n",
      "                                               0.0395099],\n",
      "                             'pre_comp_q03': [ -1.46070525,\n",
      "                                               0.66029355,\n",
      "                                               0.06603344],\n",
      "                             'pre_comp_q12': [ 0.00185588,\n",
      "                                               1.00099539,\n",
      "                                               -0.22439037],\n",
      "                             'pre_comp_q13': [ -1.02837338,\n",
      "                                               -1.10918103,\n",
      "                                               -0.80985523],\n",
      "                             'pre_comp_q22': [ -0.000190040736,\n",
      "                                               -1.34752499,\n",
      "                                               0.315152817],\n",
      "                             'pre_comp_q23': [ 0.36516325,\n",
      "                                               -0.38763126,\n",
      "                                               -0.41575102]}}\n",
      "gradient:\n",
      "{ 'capacitive_coupling_all_unify': {'strength': -0.3067819425165604},\n",
      "  'inductive_coupling_all_unify': {'strength': -12.566370000000001},\n",
      "  'q02': { 'ec': 0.4434804747756137,\n",
      "           'ej': 0.1267089558497941,\n",
      "           'el': -0.14622081146998153},\n",
      "  'q02_pulse_rampcos': { 'amp': -2.2086965168772816,\n",
      "                         'omega_d': 5.419948468635002,\n",
      "                         'phase': 0.18182549319163904,\n",
      "                         't_plateau': 0.0003573777952107349,\n",
      "                         't_ramp': 0.0007715076440344622},\n",
      "  'q03': { 'ec': -13.557677393601246,\n",
      "           'ej': 4.0855897934268475,\n",
      "           'el': -12.431920207040546},\n",
      "  'q12': { 'ec': -6.450671545556934,\n",
      "           'ej': 2.017497672846909,\n",
      "           'el': -5.852394627733166},\n",
      "  'q12_pulse_rampcos': { 'amp': -0.033454315627673,\n",
      "                         'omega_d': 4.261249160762797,\n",
      "                         'phase': 0.11020943743877394,\n",
      "                         't_plateau': -0.004077676989418231,\n",
      "                         't_ramp': -0.007391017628873898},\n",
      "  'q13': { 'ec': 6.258594744508258,\n",
      "           'ej': -2.079996230033253,\n",
      "           'el': 5.471506740061829},\n",
      "  'q22': { 'ec': -3.6920041739670455,\n",
      "           'ej': 1.3852682643226275,\n",
      "           'el': -3.3843869997766034},\n",
      "  'q22_pulse_rampcos': { 'amp': -1.2754145326607764,\n",
      "                         'omega_d': 1.9926474892396784,\n",
      "                         'phase': 0.05088425292537161,\n",
      "                         't_plateau': -0.07277049716949993,\n",
      "                         't_ramp': -0.144744165161229},\n",
      "  'single_q_compensation': { 'post_comp_q02': [ -0.04305220873590187,\n",
      "                                                -0.10800268472783614,\n",
      "                                                -0.0004724756383828765],\n",
      "                             'post_comp_q03': [ -0.04024296153032836,\n",
      "                                                0.09250102423095541,\n",
      "                                                0.04773342906392038],\n",
      "                             'post_comp_q12': [ 0.020772831600121888,\n",
      "                                                -0.11104006101152208,\n",
      "                                                -0.0007926361324744521],\n",
      "                             'post_comp_q13': [ -0.05254879462647466,\n",
      "                                                0.1328719174712205,\n",
      "                                                -0.03548128119172947],\n",
      "                             'post_comp_q22': [ 0.025936383116203388,\n",
      "                                                -0.04342921602645333,\n",
      "                                                0.00013829329107164628],\n",
      "                             'post_comp_q23': [ -0.07141244864874352,\n",
      "                                                -0.08701812697131572,\n",
      "                                                -0.2525476339244378],\n",
      "                             'pre_comp_q02': [ 0.061230351332820794,\n",
      "                                               0.006268219596307745,\n",
      "                                               0.00027801788129797246],\n",
      "                             'pre_comp_q03': [ -0.14079328145145098,\n",
      "                                               0.16294339073249375,\n",
      "                                               -0.3949979891850489],\n",
      "                             'pre_comp_q12': [ 0.08403426370829026,\n",
      "                                               -0.11317284975984876,\n",
      "                                               -0.001377586353863713],\n",
      "                             'pre_comp_q13': [ 0.0638388195785069,\n",
      "                                               0.01563048836840153,\n",
      "                                               -0.03848465146475398],\n",
      "                             'pre_comp_q22': [ 0.02596379396520243,\n",
      "                                               -0.04318233707715535,\n",
      "                                               0.00010639813381023934],\n",
      "                             'pre_comp_q23': [ 0.03038821202388675,\n",
      "                                               -0.011048212048876849,\n",
      "                                               0.06590365355608813]}}\n",
      "loss: -3.18487282367554\n",
      "step: 1\n",
      "parameters:\n",
      "{ 'capacitive_coupling_all_unify': {'strength': 0.0722576299999674},\n",
      "  'inductive_coupling_all_unify': {'strength': -0.012565370000000796},\n",
      "  'q02': { 'ec': 6.353238382300023,\n",
      "           'ej': 25.110307767500082,\n",
      "           'el': 6.344788655999932},\n",
      "  'q02_pulse_rampcos': { 'amp': 0.1812894599999955,\n",
      "                         'omega_d': 2.589344590000002,\n",
      "                         'phase': -0.242903279999945,\n",
      "                         't_plateau': 69.93608045002799,\n",
      "                         't_ramp': 29.928063880012964},\n",
      "  'q03': { 'ec': 6.2550620547,\n",
      "           'ej': 25.141139103800004,\n",
      "           'el': 5.000965058999999},\n",
      "  'q12': { 'ec': 6.2233466873999985,\n",
      "           'ej': 25.151268775200005,\n",
      "           'el': 5.601280345699998},\n",
      "  'q12_pulse_rampcos': { 'amp': 0.17872293999970107,\n",
      "                         'omega_d': 4.1998961400000026,\n",
      "                         'phase': -0.015436609999909263,\n",
      "                         't_plateau': 69.95327961999756,\n",
      "                         't_ramp': 29.945629789998648},\n",
      "  'q13': { 'ec': 6.286786210500002,\n",
      "           'ej': 25.131656050899995,\n",
      "           'el': 6.914451996200001},\n",
      "  'q22': { 'ec': 6.282847661599998,\n",
      "           'ej': 25.132995550900006,\n",
      "           'el': 7.539287440899998},\n",
      "  'q22_pulse_rampcos': { 'amp': 0.23370756999999215,\n",
      "                         'omega_d': 3.6501809100000053,\n",
      "                         'phase': -0.5000634699998036,\n",
      "                         't_plateau': 69.96302504999987,\n",
      "                         't_ramp': 29.98769403999993},\n",
      "  'single_q_compensation': { 'post_comp_q02': [ -0.015286410000232276,\n",
      "                                                0.044184189999907406,\n",
      "                                                -0.37453800002116466],\n",
      "                             'post_comp_q03': [ -1.0062967100002485,\n",
      "                                                0.1641092700001081,\n",
      "                                                -0.007976139999790504],\n",
      "                             'post_comp_q12': [ 0.0020000400004813977,\n",
      "                                                0.58680052999991,\n",
      "                                                -1.748039110012616],\n",
      "                             'post_comp_q13': [ -0.1169072600001903,\n",
      "                                                1.0997353100000753,\n",
      "                                                -1.0941258000002818],\n",
      "                             'post_comp_q22': [ -0.0014240840596144414,\n",
      "                                                -1.5985313500002303,\n",
      "                                                -0.18835374292769513],\n",
      "                             'post_comp_q23': [ 1.65405071999986,\n",
      "                                                -0.6725102500001149,\n",
      "                                                -0.0437600100000396],\n",
      "                             'pre_comp_q02': [ 0.05670905000016332,\n",
      "                                               0.026827450001595348,\n",
      "                                               0.03950890003596762],\n",
      "                             'pre_comp_q03': [ -1.460704250000071,\n",
      "                                               0.6602925500000614,\n",
      "                                               0.06603443999997469],\n",
      "                             'pre_comp_q12': [ 0.0018548800001189992,\n",
      "                                               1.0009963899999115,\n",
      "                                               -0.22438937000725903],\n",
      "                             'pre_comp_q13': [ -1.0283743799998433,\n",
      "                                               -1.1091820299993602,\n",
      "                                               -0.8098542300002599],\n",
      "                             'pre_comp_q22': [ -0.00019104073561484843,\n",
      "                                               -1.3475239900002316,\n",
      "                                               0.31515181709397777],\n",
      "                             'pre_comp_q23': [ 0.3651622500003291,\n",
      "                                               -0.3876302600009051,\n",
      "                                               -0.4157520199998483]}}\n",
      "gradient:\n",
      "{ 'capacitive_coupling_all_unify': {'strength': 0.10482803833736232},\n",
      "  'inductive_coupling_all_unify': {'strength': -12.565370000000796},\n",
      "  'q02': { 'ec': 4.609352401986676,\n",
      "           'ej': -1.2302117159246213,\n",
      "           'el': 3.549797231226524},\n",
      "  'q02_pulse_rampcos': { 'amp': -4.321642807909701,\n",
      "                         'omega_d': 1.427322706216275,\n",
      "                         'phase': 0.140673929819413,\n",
      "                         't_plateau': -0.003216670727994012,\n",
      "                         't_ramp': -0.0029878515729101184},\n",
      "  'q03': { 'ec': -4.677548891862167,\n",
      "           'ej': 1.4266089806011435,\n",
      "           'el': -4.330602663424508},\n",
      "  'q12': { 'ec': 5.151561495821057,\n",
      "           'ej': -1.5548248551095905,\n",
      "           'el': 4.578961636744389},\n",
      "  'q12_pulse_rampcos': { 'amp': -6.758932488835737,\n",
      "                         'omega_d': 4.866735929568526,\n",
      "                         'phase': 0.07026837023664591,\n",
      "                         't_plateau': -0.01104742126671375,\n",
      "                         't_ramp': -0.012781993985814227},\n",
      "  'q13': { 'ec': -9.502865058922938,\n",
      "           'ej': 3.0786728441926776,\n",
      "           'el': -8.14716494759156},\n",
      "  'q22': { 'ec': 10.838880657304626,\n",
      "           'ej': -3.3718170809787047,\n",
      "           'el': 8.801415433916185},\n",
      "  'q22_pulse_rampcos': { 'amp': 4.990213735053385,\n",
      "                         'omega_d': -0.5921462245986597,\n",
      "                         'phase': 0.010786923889322226,\n",
      "                         't_plateau': 0.4571008698646118,\n",
      "                         't_ramp': 0.9052347385548916},\n",
      "  'single_q_compensation': { 'post_comp_q02': [ -0.04307059090309089,\n",
      "                                                -0.14898068444658463,\n",
      "                                                -0.00047256810789269476],\n",
      "                             'post_comp_q03': [ -0.024088281331160033,\n",
      "                                                0.15365638738580856,\n",
      "                                                0.04578493953724768],\n",
      "                             'post_comp_q12': [ 0.020561001368234466,\n",
      "                                                0.1054100223768728,\n",
      "                                                -0.0007927764784189102],\n",
      "                             'post_comp_q13': [ -0.06311088990244684,\n",
      "                                                -0.1041573549127861,\n",
      "                                                -0.033072472023517506],\n",
      "                             'post_comp_q22': [ 0.024774161963635688,\n",
      "                                                0.1595379339811771,\n",
      "                                                0.00013790248187466872],\n",
      "                             'post_comp_q23': [ -0.04091574485756411,\n",
      "                                                -0.09112469011733992,\n",
      "                                                -0.2739033464538128],\n",
      "                             'pre_comp_q02': [ 0.06130296064246978,\n",
      "                                               -0.0346202899317305,\n",
      "                                               0.0002776779920683815],\n",
      "                             'pre_comp_q03': [ -0.04360562530867185,\n",
      "                                               0.1818952356555883,\n",
      "                                               -0.32208074809044185],\n",
      "                             'pre_comp_q12': [ 0.08301946791734438,\n",
      "                                               0.10327697747754883,\n",
      "                                               -0.0013770500845536634],\n",
      "                             'pre_comp_q13': [ 0.11641547335136693,\n",
      "                                               -0.066938995018063,\n",
      "                                               -0.13202904462595713],\n",
      "                             'pre_comp_q22': [ 0.024679152855170056,\n",
      "                                               0.1597847095212046,\n",
      "                                               0.00010698264404114699],\n",
      "                             'pre_comp_q23': [ 0.04737346304999335,\n",
      "                                               0.120703505937513,\n",
      "                                               0.06804559919823736]}}\n",
      "loss: -3.1850192847759162\n",
      "step: 2\n",
      "parameters:\n",
      "{ 'capacitive_coupling_all_unify': {'strength': 0.07225802303591318},\n",
      "  'inductive_coupling_all_unify': {'strength': -0.012564370694982277},\n",
      "  'q02': { 'ec': 6.353237577997694,\n",
      "           'ej': 25.110308438642164,\n",
      "           'el': 6.344787940552485},\n",
      "  'q02_pulse_rampcos': { 'amp': 0.18129042682667612,\n",
      "                         'omega_d': 2.5893437530419168,\n",
      "                         'phase': -0.2429042646740939,\n",
      "                         't_plateau': 69.93608111520176,\n",
      "                         't_ramp': 29.928064432708595},\n",
      "  'q03': { 'ec': 6.255062930283843,\n",
      "           'ej': 25.141138226418796,\n",
      "           'el': 5.000965936022513},\n",
      "  'q12': { 'ec': 6.223346746332991,\n",
      "           'ej': 25.151268699000436,\n",
      "           'el': 5.6012804145904544},\n",
      "  'q12_pulse_rampcos': { 'amp': 0.17872368692436003,\n",
      "                         'omega_d': 4.199895139436107,\n",
      "                         'phase': -0.01543757443619963,\n",
      "                         't_plateau': 69.95328054941416,\n",
      "                         't_ramp': 29.945630768882626},\n",
      "  'q13': { 'ec': 6.28678646345948,\n",
      "           'ej': 25.131655809354744,\n",
      "           'el': 6.914452240436086},\n",
      "  'q22': { 'ec': 6.2828471734604925,\n",
      "           'ej': 25.13299598443661,\n",
      "           'el': 7.539286986996991},\n",
      "  'q22_pulse_rampcos': { 'amp': 0.23370701524502527,\n",
      "                         'omega_d': 3.6501804800938675,\n",
      "                         'phase': -0.5000642793093042,\n",
      "                         't_plateau': 69.96302442083996,\n",
      "                         't_ramp': 29.987693411368593},\n",
      "  'single_q_compensation': { 'post_comp_q02': [ -0.015285410682274847,\n",
      "                                                0.044185185044260986,\n",
      "                                                -0.37453700073012713],\n",
      "                             'post_comp_q03': [ -1.0062957534823433,\n",
      "                                                0.16410828760873705,\n",
      "                                                -0.007977138004631843],\n",
      "                             'post_comp_q12': [ 0.0019990409739725893,\n",
      "                                                0.5868005034062224,\n",
      "                                                -1.7480381107135214],\n",
      "                             'post_comp_q13': [ -0.11690626009640448,\n",
      "                                                1.0997352420284645,\n",
      "                                                -1.0941248031392252],\n",
      "                             'post_comp_q22': [ -0.0014250819103032723,\n",
      "                                                -1.5985318917422082,\n",
      "                                                -0.18835474208772296],\n",
      "                             'post_comp_q23': [ 1.654051670735451,\n",
      "                                                -0.6725092497579671,\n",
      "                                                -0.043759009402659574],\n",
      "                             'pre_comp_q02': [ 0.05670805066253895,\n",
      "                                               0.026828062500573038,\n",
      "                                               0.039507900796888915],\n",
      "                             'pre_comp_q03': [ -1.460703390308543,\n",
      "                                               0.6602915493400233,\n",
      "                                               0.06603542890928835],\n",
      "                             'pre_comp_q12': [ 0.0018538810280393635,\n",
      "                                               1.0009963830991857,\n",
      "                                               -0.22438837071758214],\n",
      "                             'pre_comp_q13': [ -1.0283753539273084,\n",
      "                                               -1.1091814580236374,\n",
      "                                               -0.8098533287683849],\n",
      "                             'pre_comp_q22': [ -0.0001920383998789826,\n",
      "                                               -1.347524533280955,\n",
      "                                               0.3151508177415992],\n",
      "                             'pre_comp_q23': [ 0.36516126259108195,\n",
      "                                               -0.3876309395257438,\n",
      "                                               -0.4157530200119825]}}\n",
      "gradient:\n",
      "{ 'capacitive_coupling_all_unify': {'strength': 0.37683194783489327},\n",
      "  'inductive_coupling_all_unify': {'strength': -12.564370694982278},\n",
      "  'q02': { 'ec': -7.495407051823259,\n",
      "           'ej': 2.652949628231644,\n",
      "           'el': -7.030157232725677},\n",
      "  'q02_pulse_rampcos': { 'amp': -12.128708512806755,\n",
      "                         'omega_d': -2.5283093977283366,\n",
      "                         'phase': 0.10237208585731443,\n",
      "                         't_plateau': -0.01240054489789715,\n",
      "                         't_ramp': -0.010890781468303135},\n",
      "  'q03': { 'ec': 3.2915754754991204,\n",
      "           'ej': -0.9599700953993957,\n",
      "           'el': 2.941220748151496},\n",
      "  'q12': { 'ec': 8.665445617510494,\n",
      "           'ej': -2.6401402487609413,\n",
      "           'el': 7.751393713691527},\n",
      "  'q12_pulse_rampcos': { 'amp': -9.00982612151494,\n",
      "                         'omega_d': 2.6948867255035185,\n",
      "                         'phase': 0.02384912451133148,\n",
      "                         't_plateau': -0.013543392369403252,\n",
      "                         't_ramp': -0.014776885218269835},\n",
      "  'q13': { 'ec': -14.140834615892238,\n",
      "           'ej': 4.595888849539937,\n",
      "           'el': -12.152106693926367},\n",
      "  'q22': { 'ec': -1.7546071613442324,\n",
      "           'ej': 0.7448516825218735,\n",
      "           'el': -1.7414140725053793},\n",
      "  'q22_pulse_rampcos': { 'amp': -1.0090764532322074,\n",
      "                         'omega_d': -1.282481320790928,\n",
      "                         'phase': -0.009828308730368981,\n",
      "                         't_plateau': -0.3655754672949706,\n",
      "                         't_ramp': -0.7303464257112079},\n",
      "  'single_q_compensation': { 'post_comp_q02': [ -0.04318852426072234,\n",
      "                                                -0.34459856374692777,\n",
      "                                                -0.0004726193569268844],\n",
      "                             'post_comp_q03': [ -0.009478291757535118,\n",
      "                                                0.20869749786149328,\n",
      "                                                0.04403600608575101],\n",
      "                             'post_comp_q12': [ 0.020217244258629696,\n",
      "                                                0.1726989666000236,\n",
      "                                                -0.0007924495310463704],\n",
      "                             'post_comp_q13': [ -0.06813882127200088,\n",
      "                                                -0.21003210985998066,\n",
      "                                                -0.03192564248867887],\n",
      "                             'post_comp_q22': [ 0.023320914458067277,\n",
      "                                                -0.01792755286968628,\n",
      "                                                0.0001382294800765382],\n",
      "                             'post_comp_q23': [ -0.046168361069490516,\n",
      "                                                -0.09023808772110442,\n",
      "                                                -0.2677304174855563],\n",
      "                             'pre_comp_q02': [ 0.06160625983710327,\n",
      "                                               -0.22993386661802617,\n",
      "                                               0.00027723445354067237],\n",
      "                             'pre_comp_q03': [ 0.045273012544647884,\n",
      "                                               0.19898622639800578,\n",
      "                                               -0.25537322777383614],\n",
      "                             'pre_comp_q12': [ 0.08226737871308787,\n",
      "                                               0.17056679027655264,\n",
      "                                               -0.0013766075823873588],\n",
      "                             'pre_comp_q13': [ 0.12075014264968331,\n",
      "                                               -0.07373040877098447,\n",
      "                                               -0.14043498923523162],\n",
      "                             'pre_comp_q22': [ 0.023473129648083457,\n",
      "                                               -0.01767983887674518,\n",
      "                                               0.0001075211129608248],\n",
      "                             'pre_comp_q23': [ 0.05073426414843851,\n",
      "                                               0.14762712113474663,\n",
      "                                               0.06837832979228879]}}\n",
      "loss: -3.185097605293092\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "{'capacitive_coupling_all_unify': {'strength': Array(0.07225774, dtype=float64)},\n",
       " 'inductive_coupling_all_unify': {'strength': Array(-0.01256337, dtype=float64)},\n",
       " 'q02': {'ec': Array(6.35323779, dtype=float64),\n",
       "  'ej': Array(25.11030808, dtype=float64),\n",
       "  'el': Array(6.34478826, dtype=float64)},\n",
       " 'q02_pulse_rampcos': {'amp': Array(0.1812913, dtype=float64),\n",
       "  'omega_d': Array(2.58934343, dtype=float64),\n",
       "  'phase': Array(-0.24290522, dtype=float64),\n",
       "  't_plateau': Array(69.93608186, dtype=float64),\n",
       "  't_ramp': Array(29.92806516, dtype=float64)},\n",
       " 'q03': {'ec': Array(6.25506345, dtype=float64),\n",
       "  'ej': Array(25.1411377, dtype=float64),\n",
       "  'el': Array(5.00096646, dtype=float64)},\n",
       " 'q12': {'ec': Array(6.22334632, dtype=float64),\n",
       "  'ej': Array(25.15126912, dtype=float64),\n",
       "  'el': Array(5.60127999, dtype=float64)},\n",
       " 'q12_pulse_rampcos': {'amp': Array(0.17872454, dtype=float64),\n",
       "  'omega_d': Array(4.19989418, dtype=float64),\n",
       "  'phase': Array(-0.01543842, dtype=float64),\n",
       "  't_plateau': Array(69.9532815, dtype=float64),\n",
       "  't_ramp': Array(29.94563175, dtype=float64)},\n",
       " 'q13': {'ec': Array(6.28678708, dtype=float64),\n",
       "  'ej': Array(25.1316552, dtype=float64),\n",
       "  'el': Array(6.91445286, dtype=float64)},\n",
       " 'q22': {'ec': Array(6.2828469, dtype=float64),\n",
       "  'ej': Array(25.13299618, dtype=float64),\n",
       "  'el': Array(7.53928676, dtype=float64)},\n",
       " 'q22_pulse_rampcos': {'amp': Array(0.23370672, dtype=float64),\n",
       "  'omega_d': Array(3.65018053, dtype=float64),\n",
       "  'phase': Array(-0.50006478, dtype=float64),\n",
       "  't_plateau': Array(69.96302444, dtype=float64),\n",
       "  't_ramp': Array(29.98769343, dtype=float64)},\n",
       " 'single_q_compensation': {'post_comp_q02': Array([-0.01528441,  0.04418611, -0.374536  ], dtype=float64),\n",
       "  'post_comp_q03': Array([-1.0062949 ,  0.16410731, -0.00797813], dtype=float64),\n",
       "  'post_comp_q12': Array([ 0.00199804,  0.58680001, -1.74803711], dtype=float64),\n",
       "  'post_comp_q13': Array([-0.11690526,  1.09973571, -1.09412381], dtype=float64),\n",
       "  'post_comp_q22': Array([-1.42607591e-03, -1.59853224e+00, -1.88355741e-01], dtype=float64),\n",
       "  'post_comp_q23': Array([ 1.65405262, -0.67250825, -0.04375801], dtype=float64),\n",
       "  'pre_comp_q02': Array([0.05670705, 0.02682876, 0.0395069 ], dtype=float64),\n",
       "  'pre_comp_q03': Array([-1.46070294,  0.66029055,  0.0660364 ], dtype=float64),\n",
       "  'pre_comp_q12': Array([ 0.00185288,  1.0009959 , -0.22438737], dtype=float64),\n",
       "  'pre_comp_q13': Array([-1.02837634, -1.10918069, -0.80985239], dtype=float64),\n",
       "  'pre_comp_q22': Array([-1.93032678e-04, -1.34752488e+00,  3.15149819e-01], dtype=float64),\n",
       "  'pre_comp_q23': Array([ 0.36516027, -0.38763177, -0.41575402], dtype=float64)}}"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "adam_opt(infidelity,\n",
    "            params,\n",
    "            args=(target_unitary,),\n",
    "            options={\n",
    "                'adam_lr': 1e-6,\n",
    "                'steps': 3\n",
    "            })\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.13"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}