examples

Author: Phionx

docs/getting_started/examples.md

@@ -7,6 +7,11 @@ Here are some examples to help you get started with `jaxquantum`.
 Let's simulate the time dynamics of a quantum harmonic oscillator with single photon loss. 
 
 ```python 
+from jax import jit
+import jaxquantum as jqt 
+import jax.numpy as jnp
+import matplotlib.pyplot as plt
+
 N = 100
 
 omega_a = 2.0*jnp.pi*5.0

tutorials/0-examples.ipynb

@@ -10,18 +10,6 @@
     "%autoreload 2"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 7,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from jax import jit\n",
-    "import jaxquantum as jqt \n",
-    "import jax.numpy as jnp\n",
-    "import matplotlib.pyplot as plt"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -31,14 +19,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "100% |\u001b[35m██████████\u001b[0m| [00:01<00:00, 93.89%/s] \n"
+      "100% |\u001b[35m██████████\u001b[0m| [00:00<00:00, 106.67%/s]\n"
      ]
     },
     {
@@ -53,6 +41,11 @@
     }
    ],
    "source": [
+    "from jax import jit\n",
+    "import jaxquantum as jqt \n",
+    "import jax.numpy as jnp\n",
+    "import matplotlib.pyplot as plt\n",
+    "\n",
     "N = 100\n",
     "\n",
     "omega_a = 2.0*jnp.pi*5.0\n",
@@ -102,7 +95,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [