QML_HEP_GSoC_2021_Tasks_I_Quantum_Computing.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "QML-HEP GSoC 2021 Tasks I Quantum Computing.ipynb",
      "provenance": [],
      "collapsed_sections": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "code",
      "metadata": {
        "id": "OUWVHQcpAAWQ"
      },
      "source": [
        "!pip install cirq\n",
        "\n",
        "from IPython.display import clear_output\n",
        "clear_output()"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "Leyjpj2tO9zA"
      },
      "source": [
        "import cirq\n",
        "import numpy as np\n",
        "\n",
        "# visualization tools\n",
        "%matplotlib inline\n",
        "import matplotlib.pyplot as plt"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "kbNZLfmRPwHO"
      },
      "source": [
        "# Task I: Quantum Computing Part"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "IzxvrZHDP0ad"
      },
      "source": [
        "## 1). Implement a simple quantum operation with Cirq"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "SomAGSYfEGZh",
        "outputId": "e0d2caae-c4cf-4594-d91f-af096a87501c"
      },
      "source": [
        "# Initialize a circuit\n",
        "circuit = cirq.Circuit()\n",
        "# Apply a Hadamard operation on 5 qubits (one for every qubit)\n",
        "circuit.append(cirq.H(q) for q in cirq.LineQubit.range(5))\n",
        "# Apply CNOT operation on (0, 1), (1,2), (2,3), (3,4)\n",
        "circuit.append(cirq.CNOT(q, q+1) for q in cirq.LineQubit.range(4))\n",
        "# Apply SWAP (0, 4)\n",
        "circuit.append(cirq.SWAP(cirq.LineQubit(0), cirq.LineQubit(4)))\n",
        "# Apply Rotate X with pi/2 on any qubit\n",
        "circuit.append(cirq.rx(np.pi/2)(cirq.LineQubit(np.random.randint(5))))\n",
        "# Plot the circuit\n",
        "print(circuit)"
      ],
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "0: ───H───@──────────────────────×───\n",
            "          │                      │\n",
            "1: ───H───X───@──────────────────┼───\n",
            "              │                  │\n",
            "2: ───H───────X───@───Rx(0.5π)───┼───\n",
            "                  │              │\n",
            "3: ───H───────────X───@──────────┼───\n",
            "                      │          │\n",
            "4: ───H───────────────X──────────×───\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "zovsWuMobo-D"
      },
      "source": [
        "## 2). Create a circuit that is a series of small cirq.Rx rotations and plot the probability of measuring the state in the |0⟩ state"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "xTHvCHNBPBc6",
        "outputId": "24986da4-cc9d-4cec-a003-b83f0412b60b"
      },
      "source": [
        "# Initialize a circuit\n",
        "circuit = cirq.Circuit()\n",
        "# Apply a series of small cirq.Rx rotations\n",
        "rotation_angles = np.random.random(5)/10  # small random numbers for the angle of rotation\n",
        "for i in range(5):\n",
        "    circuit.append(cirq.rx(rotation_angles[i])(cirq.LineQubit(0)))\n",
        "# Add a measurement at the end of the circuit\n",
        "circuit.append(cirq.measure(cirq.LineQubit(0), key='result'))\n",
        "\n",
        "print(circuit)"
      ],
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "0: ───Rx(0.019π)───Rx(0.002π)───Rx(0.014π)───Rx(0.017π)───Rx(0.027π)───M('result')───\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 283
        },
        "id": "7diuiL7ueLMW",
        "outputId": "5dd57fac-0e96-47c6-b03c-d2dd83e0dfcb"
      },
      "source": [
        "# Initialize simulator\n",
        "s=cirq.Simulator()\n",
        "# Sample the circuit with 1000 shots\n",
        "samples=s.run(circuit, repetitions=1000)\n",
        "# Plot the probability of measuring the state in the |0⟩ state\n",
        "print(\"The probability of measuring the state in the |0⟩ state:\", samples.histogram(key=\"result\")[0]/1000)\n",
        "data = samples.measurements['result'].reshape(1000,)\n",
        "d = np.diff(np.unique(data)).min()\n",
        "left_of_first_bin = data.min() - float(d)/2\n",
        "right_of_last_bin = data.max() + float(d)/2\n",
        "plt.hist(data, np.arange(left_of_first_bin, right_of_last_bin + d, d), rwidth=0.1)\n",
        "plt.show()"
      ],
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "The probability of measuring the state in the |0⟩ state: 0.989\n"
          ],
          "name": "stdout"
        },
        {
          "output_type": "display_data",
          "data": {
            "image/png": "iVBORw0KGgoAAAANSUhEUgAAAX0AAAD4CAYAAAAAczaOAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4yLjIsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy+WH4yJAAAPBElEQVR4nO3cf6zdd13H8eeLlYH8Wsd6XWZbvTMUdcEYlpsxQoJICW7FrEuEZUSkLI1NcCAyohT9Y0YSs0VlsoQMK5t0BmFzEtfIlCzdyKKxDXcMx36Iu45tbe3ohW31x4IwefvH+WxeSruee87tuW0/z0dycz7fz/dzzvf9yW1f53s/33O+qSokSX14wXIXIEmaHENfkjpi6EtSRwx9SeqIoS9JHVmx3AU8n1WrVtX09PRylyFJJ5S77777W1U1dbh9x3XoT09PMzs7u9xlSNIJJcmjR9rn8o4kdcTQl6SOGPqS1JGjhn6SG5IcSHLfgr5XJrk9yUPt8fTWnyTXJplLcm+Scxc8Z1Mb/1CSTcdmOpKk5zPMmf6ngQsO6dsK7KyqdcDOtg1wIbCu/WwBroPBmwRwJfA64DzgymffKCRJk3PU0K+qu4AnDuneCGxv7e3AxQv6b6yBXcDKJGcBvwjcXlVPVNWTwO388BuJJOkYG3VN/8yq2t/ajwNntvZqYM+CcXtb35H6f0iSLUlmk8zOz8+PWJ4k6XDGvpBbg3szL9n9matqW1XNVNXM1NRhv1sgSRrRqKH/zbZsQ3s80Pr3AWsXjFvT+o7UL0maoFG/kbsD2ARc1R5vXdD/viSfY3DR9mBV7U/yReAPFly8fSvwkdHLHs701i8c60M855Gr3jaxY0nSqI4a+kk+C7wJWJVkL4NP4VwF3JxkM/AocEkbfhuwAZgDngYuA6iqJ5J8FPhyG/f7VXXoxWFJ0jF21NCvqnceYdf6w4wt4PIjvM4NwA2Lqk6StKT8Rq4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI4a+JHVkrNBP8sEk9ye5L8lnk7w4ydlJdieZS3JTklPb2Be17bm2f3opJiBJGt7IoZ9kNfAbwExVvQY4BbgUuBq4pqpeBTwJbG5P2Qw82fqvaeMkSRM07vLOCuBHkqwAXgLsB94M3NL2bwcubu2NbZu2f32SjHl8SdIijBz6VbUP+CPgMQZhfxC4G3iqqp5pw/YCq1t7NbCnPfeZNv6MQ183yZYks0lm5+fnRy1PknQY4yzvnM7g7P1s4MeAlwIXjFtQVW2rqpmqmpmamhr35SRJC4yzvPMW4BtVNV9V3wM+D7wBWNmWewDWAPtaex+wFqDtPw349hjHlyQt0jih/xhwfpKXtLX59cADwJ3A29uYTcCtrb2jbdP231FVNcbxJUmLNM6a/m4GF2S/AnytvdY24MPAFUnmGKzZX9+ecj1wRuu/Atg6Rt2SpBGsOPqQI6uqK4ErD+l+GDjvMGO/A7xjnONJksbjN3IlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0JakjY4V+kpVJbknyL0keTPL6JK9McnuSh9rj6W1sklybZC7JvUnOXZopSJKGNe6Z/seBv6+qnwZ+DngQ2ArsrKp1wM62DXAhsK79bAGuG/PYkqRFGjn0k5wGvBG4HqCqvltVTwEbge1t2Hbg4tbeCNxYA7uAlUnOGrlySdKijXOmfzYwD/x5knuSfCrJS4Ezq2p/G/M4cGZrrwb2LHj+3tb3A5JsSTKbZHZ+fn6M8iRJhxon9FcA5wLXVdVrgf/m/5dyAKiqAmoxL1pV26pqpqpmpqamxihPknSocUJ/L7C3qna37VsYvAl889llm/Z4oO3fB6xd8Pw1rU+SNCEjh35VPQ7sSfJTrWs98ACwA9jU+jYBt7b2DuDd7VM85wMHFywDSZImYMWYz38/8JkkpwIPA5cxeCO5Oclm4FHgkjb2NmADMAc83cZKkiZorNCvqq8CM4fZtf4wYwu4fJzjSZLG4zdyJakjhr4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI2OHfpJTktyT5G/b9tlJdieZS3JTklNb/4va9lzbPz3usSVJi7MUZ/ofAB5csH01cE1VvQp4Etjc+jcDT7b+a9o4SdIEjRX6SdYAbwM+1bYDvBm4pQ3ZDlzc2hvbNm3/+jZekjQh457p/wnw28D32/YZwFNV9Uzb3gusbu3VwB6Atv9gG/8DkmxJMptkdn5+fszyJEkLjRz6SX4JOFBVdy9hPVTVtqqaqaqZqamppXxpSereijGe+wbgoiQbgBcDrwA+DqxMsqKdza8B9rXx+4C1wN4kK4DTgG+PcXxJ0iKNfKZfVR+pqjVVNQ1cCtxRVb8C3Am8vQ3bBNza2jvaNm3/HVVVox5fkrR4x+Jz+h8Grkgyx2DN/vrWfz1wRuu/Ath6DI4tSXoe4yzvPKeqvgR8qbUfBs47zJjvAO9YiuNJkkbjN3IlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0JakjI4d+krVJ7kzyQJL7k3yg9b8yye1JHmqPp7f+JLk2yVySe5Ocu1STkCQNZ5wz/WeAD1XVOcD5wOVJzgG2Ajurah2ws20DXAisaz9bgOvGOLYkaQQjh35V7a+qr7T2fwIPAquBjcD2Nmw7cHFrbwRurIFdwMokZ41cuSRp0ZZkTT/JNPBaYDdwZlXtb7seB85s7dXAngVP29v6Dn2tLUlmk8zOz88vRXmSpGbs0E/yMuCvgd+sqv9YuK+qCqjFvF5VbauqmaqamZqaGrc8SdICY4V+khcyCPzPVNXnW/c3n122aY8HWv8+YO2Cp69pfZKkCRnn0zsBrgcerKqPLdi1A9jU2puAWxf0v7t9iud84OCCZSBJ0gSsGOO5bwB+Ffhakq+2vt8BrgJuTrIZeBS4pO27DdgAzAFPA5eNcWxJ0ghGDv2q+gcgR9i9/jDjC7h81ONJksbnN3IlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0JakjKyZ9wCQXAB8HTgE+VVVXTboGSVqs6a1fmOjxHrnqbcfkdSd6pp/kFOATwIXAOcA7k5wzyRokqWeTXt45D5irqoer6rvA54CNE65Bkro16eWd1cCeBdt7gdctHJBkC7Clbf5Xkq9PqLZnrQK+tdgn5epjUMmxMdL8ThDO7cR1Ms9vOTLlJ460Y+Jr+kdTVduAbct1/CSzVTWzXMc/1k7m+Tm3E9fJPL/jbW6TXt7ZB6xdsL2m9UmSJmDSof9lYF2Ss5OcClwK7JhwDZLUrYku71TVM0neB3yRwUc2b6iq+ydZwxCWbWlpQk7m+Tm3E9fJPL/jam6pquWuQZI0IX4jV5I6YuhLUke6Df0kFyT5epK5JFsPs/9FSW5q+3cnmZ58laMZYm5XJHkgyb1JdiY54md6j0dHm9+Ccb+cpJIcNx+XO5ph5pbkkvb7uz/JX066xlEN8e/yx5PcmeSe9m9zw3LUOYokNyQ5kOS+I+xPkmvb3O9Ncu6ka3xOVXX3w+Ai8r8BPwmcCvwzcM4hY34d+GRrXwrctNx1L+HcfgF4SWu/90SZ27Dza+NeDtwF7AJmlrvuJfzdrQPuAU5v2z+63HUv4dy2Ae9t7XOAR5a77kXM743AucB9R9i/Afg7IMD5wO7lqrXXM/1hbgexEdje2rcA65NkgjWO6qhzq6o7q+rptrmLwfclThTD3srjo8DVwHcmWdyYhpnbrwGfqKonAarqwIRrHNUwcyvgFa19GvDvE6xvLFV1F/DE8wzZCNxYA7uAlUnOmkx1P6jX0D/c7SBWH2lMVT0DHATOmEh14xlmbgttZnAGcqI46vzan85rq2qyt0Uc3zC/u1cDr07yj0l2tbvWngiGmdvvAe9Kshe4DXj/ZEqbiMX+vzxmjrvbMGhykrwLmAF+frlrWSpJXgB8DHjPMpdyrKxgsMTzJgZ/od2V5Ger6qllrWppvBP4dFX9cZLXA3+R5DVV9f3lLuxk0uuZ/jC3g3huTJIVDP7c/PZEqhvPULe6SPIW4HeBi6rqfyZU21I42vxeDrwG+FKSRxisn+44QS7mDvO72wvsqKrvVdU3gH9l8CZwvBtmbpuBmwGq6p+AFzO4WdnJ4Li5BU2voT/M7SB2AJta++3AHdWuyBznjjq3JK8F/pRB4J8oa8LPet75VdXBqlpVVdNVNc3gmsVFVTW7POUuyjD/Lv+GwVk+SVYxWO55eJJFjmiYuT0GrAdI8jMMQn9+olUeOzuAd7dP8ZwPHKyq/ctRSJfLO3WE20Ek+X1gtqp2ANcz+PNyjsEFmkuXr+LhDTm3PwReBvxVuzb9WFVdtGxFL8KQ8zshDTm3LwJvTfIA8L/Ab1XVcf8X6JBz+xDwZ0k+yOCi7ntOkBMtknyWwZvxqnZN4krghQBV9UkG1yg2AHPA08Bly1Opt2GQpK70urwjSV0y9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JH/g9upt4rYeVUkgAAAABJRU5ErkJggg==\n",
            "text/plain": [
              "<Figure size 432x288 with 1 Axes>"
            ]
          },
          "metadata": {
            "tags": [],
            "needs_background": "light"
          }
        }
      ]
    }
  ]
}