diff --git a/brainunit/_base.py b/brainunit/_base.py
index 577e281..8640d87 100644
--- a/brainunit/_base.py
+++ b/brainunit/_base.py
@@ -32,6 +32,7 @@
   'Quantity',
   'Unit',
   'UnitRegistry',
+  'Dimension',
   'DIMENSIONLESS',
   'DimensionMismatchError',
   'get_or_create_dimension',
@@ -458,6 +459,7 @@ def get_or_create_dimension(*args, **kwds):
     return new_dim
 
 
+'''The dimensionless unit, used for quantities without a unit.'''
 DIMENSIONLESS = Dimension((0, 0, 0, 0, 0, 0, 0))
 _dimensions = {(0, 0, 0, 0, 0, 0, 0): DIMENSIONLESS}
 
diff --git a/docs/tutorials/physical_units.ipynb b/docs/tutorials/physical_units.ipynb
index 444756d..f6ebb36 100644
--- a/docs/tutorials/physical_units.ipynb
+++ b/docs/tutorials/physical_units.ipynb
@@ -31,9 +31,9 @@
    "metadata": {},
    "source": [
     "## Importing units\n",
-    "Braincore generates standard names for units, combining the unit name (e.g. “siemens”) with a prefixes (e.g. “m”), and also generates squared and cubed versions by appending a number. For example, the units “msiemens”, “siemens2”, “usiemens3” are all predefined. You can import these units from the package `briancore.units` – accordingly, an `from braincore.units import *` will result in everything being imported.\n",
+    "Braincore generates standard names for units, combining the unit name (e.g. “siemens”) with a prefixes (e.g. “m”), and also generates squared and cubed versions by appending a number. For example, the units “msiemens”, “siemens2”, “usiemens3” are all predefined. You can import these units from the package `brianunit` – accordingly, an `from brainunit import *` will result in everything being imported.\n",
     "\n",
-    "We recommend importing only the units you need, to have a cleaner namespace. For example, `import braincore.units as U` and then using `bu.msiemens` instead of `msiemens`."
+    "We recommend importing only the units you need, to have a cleaner namespace. For example, `import brainunit as bu` and then using `bu.msiemens` instead of `msiemens`."
    ]
   },
   {
@@ -45,10 +45,10 @@
      "start_time": "2024-06-08T08:18:06.566328Z"
     }
    },
+   "outputs": [],
    "source": [
     "import brainunit as bu"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -62,31 +62,65 @@
    "cell_type": "code",
    "execution_count": 2,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "20. * msecond"
+      ]
+     },
+     "execution_count": 2,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "tau = 20 * bu.ms\n",
     "tau"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 3,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([10., 20., 30.], dtype=float32) * hertz"
+      ]
+     },
+     "execution_count": 3,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "rates = [10, 20, 30] * bu.Hz\n",
     "rates"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 4,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[10., 20., 30.],\n",
+       "           [20., 30., 40.]], dtype=float32) * hertz"
+      ]
+     },
+     "execution_count": 4,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "rates = [[10, 20, 30], [20, 30, 40]] * bu.Hz\n",
     "rates"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -99,25 +133,41 @@
    "cell_type": "code",
    "execution_count": 5,
    "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Cannot calculate ... += 1, units do not match (units are s and 1).\n"
+     ]
+    }
+   ],
    "source": [
     "try:\n",
     "    tau += 1  # ms? second?\n",
     "except Exception as e:\n",
     "    print(e)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 6,
    "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Cannot calculate 3.0 + 3.0, units do not match (units are kg and A).\n"
+     ]
+    }
+   ],
    "source": [
     "try:\n",
     "    3 * bu.kgram + 3 * bu.amp\n",
     "except Exception as e:\n",
     "    print(e)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -151,37 +201,83 @@
    "cell_type": "code",
    "execution_count": 7,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[10., 20., 30.],\n",
+       "           [20., 30., 40.]], dtype=float32) * hertz"
+      ]
+     },
+     "execution_count": 7,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "rates"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 8,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Array([[10., 20., 30.],\n",
+       "       [20., 30., 40.]], dtype=float32)"
+      ]
+     },
+     "execution_count": 8,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "rates.value"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 9,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "second ** -1"
+      ]
+     },
+     "execution_count": 9,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "rates.unit"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 10,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(2, (2, 3), 6, dtype('float32'))"
+      ]
+     },
+     "execution_count": 10,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "rates.ndim, rates.shape, rates.size, rates.dtype"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -195,12 +291,12 @@
    "cell_type": "code",
    "execution_count": 11,
    "metadata": {},
+   "outputs": [],
    "source": [
     "import jax.numpy as jnp\n",
     "import brainstate as bst\n",
     "bst.environ.set(precision=64) # we recommend using 64-bit precision for better numerical stability"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -215,10 +311,21 @@
    "cell_type": "code",
    "execution_count": 12,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "5. * msecond"
+      ]
+     },
+     "execution_count": 12,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "5 * bu.ms"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -231,10 +338,21 @@
    "cell_type": "code",
    "execution_count": 13,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "5. * msecond"
+      ]
+     },
+     "execution_count": 13,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "jnp.float64(5) * bu.ms"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -247,10 +365,21 @@
    "cell_type": "code",
    "execution_count": 14,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([1., 2., 3.]) * msecond"
+      ]
+     },
+     "execution_count": 14,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "jnp.array([1, 2, 3]) * bu.ms"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -263,10 +392,21 @@
    "cell_type": "code",
    "execution_count": 15,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([1., 2., 3.]) * msecond"
+      ]
+     },
+     "execution_count": 15,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "[1, 2, 3] * bu.ms"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -286,10 +426,21 @@
    "cell_type": "code",
    "execution_count": 16,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Quantity(5.)"
+      ]
+     },
+     "execution_count": 16,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.Quantity(5)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -302,10 +453,21 @@
    "cell_type": "code",
    "execution_count": 18,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "5. * msecond"
+      ]
+     },
+     "execution_count": 18,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.Quantity(5, unit=bu.ms)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -318,10 +480,21 @@
    "cell_type": "code",
    "execution_count": 19,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([1., 2., 3.]) * msecond"
+      ]
+     },
+     "execution_count": 19,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.Quantity(jnp.array([1, 2, 3]), unit=bu.ms)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -334,10 +507,21 @@
    "cell_type": "code",
    "execution_count": 20,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([1., 2., 3.]) * msecond"
+      ]
+     },
+     "execution_count": 20,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.Quantity([1, 2, 3], unit=bu.ms)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -350,10 +534,21 @@
    "cell_type": "code",
    "execution_count": 21,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([0.5, 1. ]) * second"
+      ]
+     },
+     "execution_count": 21,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.Quantity([500 * bu.ms, 1 * bu.second])"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -366,10 +561,21 @@
    "cell_type": "code",
    "execution_count": 22,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([0.5, 1. ]) * second"
+      ]
+     },
+     "execution_count": 22,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.Quantity.with_units(jnp.array([0.5, 1]), second=1)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -383,28 +589,61 @@
    "cell_type": "code",
    "execution_count": 23,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Quantity(ArrayImpl([1., 2., 3.]))"
+      ]
+     },
+     "execution_count": 23,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.Quantity([1, 2, 3])"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 24,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Quantity(ArrayImpl([1., 2., 3.]))"
+      ]
+     },
+     "execution_count": 24,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.Quantity(jnp.array([1, 2, 3]))"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 25,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Quantity(ArrayImpl([], dtype=float64))"
+      ]
+     },
+     "execution_count": 25,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.Quantity([])"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -418,37 +657,61 @@
    "cell_type": "code",
    "execution_count": 26,
    "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "All elements must have the same unit\n"
+     ]
+    }
+   ],
    "source": [
     "try:\n",
     "    bu.Quantity([500 * bu.ms, 1])\n",
     "except Exception as e:\n",
     "    print(e)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 27,
    "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Value 'some' with dtype <U4 is not a valid JAX array type. Only arrays of numeric types are supported by JAX.\n"
+     ]
+    }
+   ],
    "source": [
     "try:\n",
     "    bu.Quantity([\"some\", \"nonsense\"])\n",
     "except Exception as e:\n",
     "    print(e)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 28,
    "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "All elements must have the same unit\n"
+     ]
+    }
+   ],
    "source": [
     "try:\n",
     "    bu.Quantity([500 * bu.ms, 1 * bu.volt])\n",
     "except Exception as e:\n",
     "    print(e)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -462,41 +725,85 @@
    "cell_type": "code",
    "execution_count": 29,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([0., 1., 2., 3.]) * mvolt"
+      ]
+     },
+     "execution_count": 29,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a = [20, 30, 40, 50] * bu.mV\n",
     "b = jnp.arange(4) * bu.mV\n",
     "b"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 30,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([20., 29., 38., 47.]) * mvolt"
+      ]
+     },
+     "execution_count": 30,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "c = a - b\n",
     "c"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 31,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([0., 1., 4., 9.]) * mvolt2"
+      ]
+     },
+     "execution_count": 31,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b = b ** 2\n",
     "b"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 32,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Array([ True,  True, False, False], dtype=bool)"
+      ]
+     },
+     "execution_count": 32,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a < 35 * bu.mV"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -509,31 +816,67 @@
    "cell_type": "code",
    "execution_count": 33,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[ 2.,  6.],\n",
+       "           [12., 20.]]) * mvolt2"
+      ]
+     },
+     "execution_count": 33,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "A = jnp.array([[1, 2], [3, 4]]) * bu.mV\n",
     "B = jnp.array([[2, 3], [4, 5]]) * bu.mV\n",
     "\n",
     "A * B"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 34,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[10., 13.],\n",
+       "           [22., 29.]]) * mvolt2"
+      ]
+     },
+     "execution_count": 34,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "A @ B"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 35,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[10., 13.],\n",
+       "           [22., 29.]]) * mvolt2"
+      ]
+     },
+     "execution_count": 35,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "A.dot(B)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -546,13 +889,24 @@
    "cell_type": "code",
    "execution_count": 36,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'float64'"
+      ]
+     },
+     "execution_count": 36,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a = jnp.ones(3, dtype=jnp.int32) * bu.mV\n",
     "b = jnp.linspace(0, jnp.pi, 3) * bu.mV\n",
     "\n",
     "b.dtype.name"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -563,11 +917,22 @@
      "start_time": "2024-06-08T08:17:11.917832Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'float64'"
+      ]
+     },
+     "execution_count": 37,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "c = a + b\n",
     "c.dtype.name"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -585,11 +950,23 @@
      "start_time": "2024-06-08T08:17:12.915631Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[0.6437081 , 0.66346343, 0.92401136],\n",
+       "           [0.07923831, 0.78219192, 0.28288394]]) * mvolt"
+      ]
+     },
+     "execution_count": 38,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a = bst.random.random((2, 3)) * bu.mV\n",
     "a"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -600,10 +977,21 @@
      "start_time": "2024-06-08T08:17:13.061209Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "3.37549707 * mvolt"
+      ]
+     },
+     "execution_count": 39,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.sum()"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -614,10 +1002,21 @@
      "start_time": "2024-06-08T08:17:13.154985Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "79.23830773 * uvolt"
+      ]
+     },
+     "execution_count": 40,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.min()"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -628,10 +1027,21 @@
      "start_time": "2024-06-08T08:17:13.245139Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "0.92401136 * mvolt"
+      ]
+     },
+     "execution_count": 41,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.max()"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -649,11 +1059,24 @@
      "start_time": "2024-06-08T08:17:13.303347Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[ 0.,  1.,  2.,  3.],\n",
+       "           [ 4.,  5.,  6.,  7.],\n",
+       "           [ 8.,  9., 10., 11.]]) * mvolt"
+      ]
+     },
+     "execution_count": 42,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b = jnp.arange(12).reshape(3, 4) * bu.mV\n",
     "b"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -664,10 +1087,21 @@
      "start_time": "2024-06-08T08:17:13.423512Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([12., 15., 18., 21.]) * mvolt"
+      ]
+     },
+     "execution_count": 43,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b.sum(axis=0) # sum of each column"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -678,10 +1112,21 @@
      "start_time": "2024-06-08T08:17:13.493763Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([0., 4., 8.]) * mvolt"
+      ]
+     },
+     "execution_count": 44,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b.min(axis=1) # min of each row"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -692,10 +1137,23 @@
      "start_time": "2024-06-08T08:17:13.623775Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[ 0.,  1.,  3.,  6.],\n",
+       "           [ 4.,  9., 15., 22.],\n",
+       "           [ 8., 17., 27., 38.]]) * mvolt"
+      ]
+     },
+     "execution_count": 45,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b.cumsum(axis=1) # cumulative sum along each row"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -715,11 +1173,22 @@
      "start_time": "2024-06-08T08:17:13.724137Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Quantity(ArrayImpl([0., 1., 2.]))"
+      ]
+     },
+     "execution_count": 46,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "B = bu.Quantity(jnp.arange(3))\n",
     "B"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -730,10 +1199,21 @@
      "start_time": "2024-06-08T08:17:13.835550Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Array([1.        , 2.71828183, 7.3890561 ], dtype=float64)"
+      ]
+     },
+     "execution_count": 47,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.exp(B)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -744,10 +1224,21 @@
      "start_time": "2024-06-08T08:17:13.899481Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Array([0.        , 1.        , 1.41421356], dtype=float64)"
+      ]
+     },
+     "execution_count": 48,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.sqrt(B)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -758,11 +1249,22 @@
      "start_time": "2024-06-08T08:17:14.017580Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "Quantity(ArrayImpl([0., 2., 4.]))"
+      ]
+     },
+     "execution_count": 49,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "C = bu.Quantity(jnp.arange(3))\n",
     "bu.math.add(B, C)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -781,11 +1283,22 @@
      "start_time": "2024-06-08T08:17:14.077645Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([  0.,   1.,   8.,  27.,  64., 125., 216., 343., 512., 729.]) * mvolt"
+      ]
+     },
+     "execution_count": 50,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a = jnp.arange(10) ** 3 * bu.mV\n",
     "a"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -796,10 +1309,21 @@
      "start_time": "2024-06-08T08:17:14.233917Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "8. * mvolt"
+      ]
+     },
+     "execution_count": 51,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a[2]"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -810,10 +1334,21 @@
      "start_time": "2024-06-08T08:17:14.323023Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([ 8., 27., 64.]) * mvolt"
+      ]
+     },
+     "execution_count": 52,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a[2:5]"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -831,13 +1366,25 @@
      "start_time": "2024-06-08T08:17:14.423077Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([1000.,    1., 1000.,   27., 1000.,  125.,  216.,  343.,  512.,\n",
+       "            729.]) * mvolt"
+      ]
+     },
+     "execution_count": 53,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "# equivalent to a[0:6:2] = 1000;\n",
     "# from start to position 6, exclusive, set every 2nd element to 1000\n",
     "a[:6:2] = 1000 * bu.mV\n",
     "a"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -848,10 +1395,22 @@
      "start_time": "2024-06-08T08:17:14.484540Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([ 729.,  512.,  343.,  216.,  125., 1000.,   27., 1000.,    1.,\n",
+       "           1000.]) * mvolt"
+      ]
+     },
+     "execution_count": 54,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a[::-1] # reversed a"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -862,11 +1421,28 @@
      "start_time": "2024-06-08T08:17:14.626722Z"
     }
    },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "1. m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n",
+      "0.1 m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n",
+      "1. m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n",
+      "0.3 m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n",
+      "1. m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n",
+      "0.5 m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n",
+      "0.6 m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n",
+      "0.7 m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n",
+      "0.8 m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n",
+      "0.9 m^0.6666666666666666 kg^0.3333333333333333 s^-1.0 A^-0.3333333333333333\n"
+     ]
+    }
+   ],
    "source": [
     "for i in a:\n",
     "    print(i**(1 / 3.))"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -884,13 +1460,28 @@
      "start_time": "2024-06-08T08:17:14.675521Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[ 0.,  1.,  2.,  3.],\n",
+       "           [10., 11., 12., 13.],\n",
+       "           [20., 21., 22., 23.],\n",
+       "           [30., 31., 32., 33.],\n",
+       "           [40., 41., 42., 43.]]) * mvolt"
+      ]
+     },
+     "execution_count": 56,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "def f(x, y):\n",
     "    return 10 * x + y\n",
     "b = jnp.fromfunction(f, (5, 4), dtype=jnp.int32) * bu.mV\n",
     "b"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -901,10 +1492,21 @@
      "start_time": "2024-06-08T08:17:14.810619Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "23. * mvolt"
+      ]
+     },
+     "execution_count": 57,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b[2, 3]"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -915,10 +1517,21 @@
      "start_time": "2024-06-08T08:17:14.890825Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([ 1., 11., 21., 31., 41.]) * mvolt"
+      ]
+     },
+     "execution_count": 58,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b[0:5, 1]  # each row in the second column of b"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -929,10 +1542,21 @@
      "start_time": "2024-06-08T08:17:15.010562Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([ 1., 11., 21., 31., 41.]) * mvolt"
+      ]
+     },
+     "execution_count": 59,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b[:, 1]  # equivalent to the previous example"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -943,10 +1567,22 @@
      "start_time": "2024-06-08T08:17:15.069946Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[10., 11., 12., 13.],\n",
+       "           [20., 21., 22., 23.]]) * mvolt"
+      ]
+     },
+     "execution_count": 60,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b[1:3, :]  # each column in the second and third row of b"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -964,10 +1600,21 @@
      "start_time": "2024-06-08T08:17:15.169438Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([40., 41., 42., 43.]) * mvolt"
+      ]
+     },
+     "execution_count": 61,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b[-1]"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -990,11 +1637,22 @@
      "start_time": "2024-06-08T08:17:15.224044Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(2, 2, 3)"
+      ]
+     },
+     "execution_count": 62,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "c = jnp.array([[[0, 1, 2], [10, 12, 13]], [[100, 101, 102], [110, 112, 113]]]) * bu.mV # a 3D array (two stacked 2D arrays)\n",
     "c.shape"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1005,10 +1663,22 @@
      "start_time": "2024-06-08T08:17:15.364283Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[100., 101., 102.],\n",
+       "           [110., 112., 113.]]) * mvolt"
+      ]
+     },
+     "execution_count": 63,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "c[1, ...] # same as c[1, :, :] or c[1]"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1019,10 +1689,22 @@
      "start_time": "2024-06-08T08:17:15.467641Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[  2.,  13.],\n",
+       "           [102., 113.]]) * mvolt"
+      ]
+     },
+     "execution_count": 64,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "c[..., 2] # same as c[:, :, 2]"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1040,11 +1722,23 @@
      "start_time": "2024-06-08T08:17:15.568297Z"
     }
    },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "[0. 1. 2. 3.] mV\n",
+      "[10. 11. 12. 13.] mV\n",
+      "[20. 21. 22. 23.] mV\n",
+      "[30. 31. 32. 33.] mV\n",
+      "[40. 41. 42. 43.] mV\n"
+     ]
+    }
+   ],
    "source": [
     "for row in b:\n",
     "    print(row)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1064,11 +1758,24 @@
      "start_time": "2024-06-08T08:17:15.628497Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[0., 8., 8., 9.],\n",
+       "           [1., 6., 2., 0.],\n",
+       "           [2., 0., 5., 0.]]) * mvolt"
+      ]
+     },
+     "execution_count": 66,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a = jnp.floor(10 * bst.random.random((3, 4))) * bu.mV\n",
     "a"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1079,10 +1786,21 @@
      "start_time": "2024-06-08T08:17:15.774172Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(3, 4)"
+      ]
+     },
+     "execution_count": 67,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.shape"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1100,10 +1818,21 @@
      "start_time": "2024-06-08T08:17:15.851908Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([0., 8., 8., 9., 1., 6., 2., 0., 2., 0., 5., 0.]) * mvolt"
+      ]
+     },
+     "execution_count": 68,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.ravel()  # returns the array, flattened"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1114,10 +1843,26 @@
      "start_time": "2024-06-08T08:17:15.972194Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[0., 8.],\n",
+       "           [8., 9.],\n",
+       "           [1., 6.],\n",
+       "           [2., 0.],\n",
+       "           [2., 0.],\n",
+       "           [5., 0.]]) * mvolt"
+      ]
+     },
+     "execution_count": 69,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.reshape(6, 2) # returns the array with a modified shape"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1128,10 +1873,24 @@
      "start_time": "2024-06-08T08:17:16.058519Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[0., 1., 2.],\n",
+       "           [8., 6., 0.],\n",
+       "           [8., 2., 5.],\n",
+       "           [9., 0., 0.]]) * mvolt"
+      ]
+     },
+     "execution_count": 70,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.T  # returns the array, transposed"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1142,10 +1901,21 @@
      "start_time": "2024-06-08T08:17:16.160362Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(4, 3)"
+      ]
+     },
+     "execution_count": 71,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.T.shape"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1156,10 +1926,21 @@
      "start_time": "2024-06-08T08:17:16.224359Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(3, 4)"
+      ]
+     },
+     "execution_count": 72,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.shape"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1186,10 +1967,23 @@
      "start_time": "2024-06-08T08:17:16.311537Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[0., 8., 8., 9.],\n",
+       "           [1., 6., 2., 0.],\n",
+       "           [2., 0., 5., 0.]]) * mvolt"
+      ]
+     },
+     "execution_count": 73,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a.reshape(3, -1)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1208,11 +2002,23 @@
      "start_time": "2024-06-08T08:17:16.403141Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[9., 1.],\n",
+       "           [0., 0.]]) * mvolt"
+      ]
+     },
+     "execution_count": 74,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a = jnp.floor(10 * bst.random.random((2, 2))) * bu.mV\n",
     "a"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1223,11 +2029,23 @@
      "start_time": "2024-06-08T08:17:16.565752Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[6., 3.],\n",
+       "           [2., 9.]]) * mvolt"
+      ]
+     },
+     "execution_count": 75,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "b = jnp.floor(10 * bst.random.random((2, 2))) * bu.mV\n",
     "b"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1238,10 +2056,24 @@
      "start_time": "2024-06-08T08:17:16.658982Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[9., 1.],\n",
+       "           [0., 0.],\n",
+       "           [6., 3.],\n",
+       "           [2., 9.]]) * mvolt"
+      ]
+     },
+     "execution_count": 76,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.vstack((a, b))"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1252,10 +2084,22 @@
      "start_time": "2024-06-08T08:17:16.732009Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[9., 1., 6., 3.],\n",
+       "           [0., 0., 2., 9.]]) * mvolt"
+      ]
+     },
+     "execution_count": 77,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.hstack((a, b))"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1273,10 +2117,22 @@
      "start_time": "2024-06-08T08:17:16.835479Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[9., 1., 6., 3.],\n",
+       "           [0., 0., 2., 9.]]) * mvolt"
+      ]
+     },
+     "execution_count": 78,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.column_stack((a, b)) # with 2D arrays"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1287,12 +2143,24 @@
      "start_time": "2024-06-08T08:17:16.919138Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[4., 2.],\n",
+       "           [2., 8.]]) * mvolt"
+      ]
+     },
+     "execution_count": 79,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a = jnp.array([4., 2.]) * bu.mV\n",
     "b = jnp.array([2., 8.]) * bu.mV\n",
     "bu.math.column_stack((a, b)) # returns a 2D array"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1303,10 +2171,21 @@
      "start_time": "2024-06-08T08:17:17.074655Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([4., 2., 2., 8.]) * mvolt"
+      ]
+     },
+     "execution_count": 80,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.hstack((a, b)) # the result is different"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1317,10 +2196,22 @@
      "start_time": "2024-06-08T08:17:17.137485Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[4.],\n",
+       "           [2.]]) * mvolt"
+      ]
+     },
+     "execution_count": 81,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a[:, jnp.newaxis] # view `a` as a 2D column vector"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1331,10 +2222,22 @@
      "start_time": "2024-06-08T08:17:17.226299Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[4., 2.],\n",
+       "           [2., 8.]]) * mvolt"
+      ]
+     },
+     "execution_count": 82,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.column_stack((a[:, jnp.newaxis], b[:, jnp.newaxis]))"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1345,10 +2248,22 @@
      "start_time": "2024-06-08T08:17:17.284862Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[4., 2.],\n",
+       "           [2., 8.]]) * mvolt"
+      ]
+     },
+     "execution_count": 83,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.hstack((a[:, jnp.newaxis], b[:, jnp.newaxis])) # the result is the same"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1367,11 +2282,23 @@
      "start_time": "2024-06-08T08:17:17.378675Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[3., 6., 6., 8., 0., 0., 5., 8., 9., 1., 5., 1.],\n",
+       "           [7., 3., 0., 2., 3., 9., 3., 7., 3., 4., 0., 8.]]) * mvolt"
+      ]
+     },
+     "execution_count": 84,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "a = jnp.floor(10 * bst.random.random((2, 12))) * bu.mV\n",
     "a"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1382,10 +2309,26 @@
      "start_time": "2024-06-08T08:17:17.454110Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[ArrayImpl([[3., 6., 6., 8.],\n",
+       "            [7., 3., 0., 2.]]) * mvolt,\n",
+       " ArrayImpl([[0., 0., 5., 8.],\n",
+       "            [3., 9., 3., 7.]]) * mvolt,\n",
+       " ArrayImpl([[9., 1., 5., 1.],\n",
+       "            [3., 4., 0., 8.]]) * mvolt]"
+      ]
+     },
+     "execution_count": 85,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.hsplit(a, 3) # Split a into 3"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1396,10 +2339,26 @@
      "start_time": "2024-06-08T08:17:17.590257Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[ArrayImpl([[3., 6., 6.],\n",
+       "            [7., 3., 0.]]) * mvolt,\n",
+       " ArrayImpl([[8.],\n",
+       "            [2.]]) * mvolt,\n",
+       " ArrayImpl([[0., 0., 5., 8., 9., 1., 5., 1.],\n",
+       "            [3., 9., 3., 7., 3., 4., 0., 8.]]) * mvolt]"
+      ]
+     },
+     "execution_count": 86,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.math.hsplit(a, (3, 4)) # Split `a` after the third and the fourth column"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1427,11 +2386,22 @@
      "start_time": "2024-06-08T08:17:17.651627Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "3. * volt"
+      ]
+     },
+     "execution_count": 87,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q = 3. * bu.volt\n",
     "q"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1442,10 +2412,21 @@
      "start_time": "2024-06-08T08:17:17.774669Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'3000. mV'"
+      ]
+     },
+     "execution_count": 88,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "bu.in_unit(q, bu.mvolt)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1464,11 +2445,22 @@
      "start_time": "2024-06-08T08:17:17.836718Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "5. * mvolt"
+      ]
+     },
+     "execution_count": 89,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q = 5. * bu.mV\n",
     "q"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1479,11 +2471,22 @@
      "start_time": "2024-06-08T08:17:17.932601Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "-5. * mvolt"
+      ]
+     },
+     "execution_count": 90,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q = -q\n",
     "q"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1494,11 +2497,22 @@
      "start_time": "2024-06-08T08:17:18.005084Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "-5. * mvolt"
+      ]
+     },
+     "execution_count": 91,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q = +q\n",
     "q"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1509,11 +2523,22 @@
      "start_time": "2024-06-08T08:17:18.139911Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "5. * mvolt"
+      ]
+     },
+     "execution_count": 92,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q = abs(q)\n",
     "q"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1524,12 +2549,23 @@
      "start_time": "2024-06-08T08:17:18.184689Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(Quantity(5.), Quantity(-6.))"
+      ]
+     },
+     "execution_count": 93,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 = bu.Quantity(0b101)\n",
     "q2 = bu.Quantity(-0b110)\n",
     "q1, q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1540,10 +2576,21 @@
      "start_time": "2024-06-08T08:17:18.286532Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(Quantity(-6.), Quantity(5.))"
+      ]
+     },
+     "execution_count": 94,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "~q1, ~q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1562,12 +2609,24 @@
      "start_time": "2024-06-08T08:17:18.335369Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(ArrayImpl([10., 12., 14., 16., 18.]) * mvolt,\n",
+       " ArrayImpl([ 8., 12., 16., 20., 24.]) * mvolt)"
+      ]
+     },
+     "execution_count": 95,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 = jnp.arange(10, 20, 2) * bu.mV\n",
     "q2 = jnp.arange(8, 27, 4) * bu.mV\n",
     "q1, q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1578,10 +2637,22 @@
      "start_time": "2024-06-08T08:17:18.448183Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(Array([False,  True, False, False, False], dtype=bool),\n",
+       " Array([ True, False,  True,  True,  True], dtype=bool))"
+      ]
+     },
+     "execution_count": 96,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 == q2, q1 != q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1592,10 +2663,22 @@
      "start_time": "2024-06-08T08:17:18.523328Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(Array([False, False,  True,  True,  True], dtype=bool),\n",
+       " Array([False,  True,  True,  True,  True], dtype=bool))"
+      ]
+     },
+     "execution_count": 97,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 < q2, q1 <= q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1606,10 +2689,22 @@
      "start_time": "2024-06-08T08:17:18.644318Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(Array([ True, False, False, False, False], dtype=bool),\n",
+       " Array([ True,  True, False, False, False], dtype=bool))"
+      ]
+     },
+     "execution_count": 98,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 > q2, q1 >= q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1628,12 +2723,23 @@
      "start_time": "2024-06-08T08:17:18.733415Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(ArrayImpl([1., 2., 3.]) * mvolt, ArrayImpl([2., 3., 4.]) * mvolt)"
+      ]
+     },
+     "execution_count": 99,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 = jnp.array([1, 2, 3]) * bu.mV\n",
     "q2 = jnp.array([2, 3, 4]) * bu.mV\n",
     "q1, q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1644,10 +2750,21 @@
      "start_time": "2024-06-08T08:17:18.840386Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(ArrayImpl([3., 5., 7.]) * mvolt, ArrayImpl([-1., -1., -1.]) * mvolt)"
+      ]
+     },
+     "execution_count": 100,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 + q2, q1 - q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1658,10 +2775,21 @@
      "start_time": "2024-06-08T08:17:18.927264Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([ 2.,  6., 12.]) * mvolt2"
+      ]
+     },
+     "execution_count": 101,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 * q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1672,10 +2800,23 @@
      "start_time": "2024-06-08T08:17:19.045255Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(Array([0.5       , 0.66666667, 0.75      ], dtype=float64),\n",
+       " Array([0., 0., 0.], dtype=float64),\n",
+       " ArrayImpl([1., 2., 3.]) * mvolt)"
+      ]
+     },
+     "execution_count": 102,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 / q2, q1 // q2, q1 % q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1686,10 +2827,21 @@
      "start_time": "2024-06-08T08:17:19.125338Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(Array([0., 0., 0.], dtype=float64), ArrayImpl([1., 2., 3.]) * mvolt)"
+      ]
+     },
+     "execution_count": 103,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "divmod(q1, q2)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1700,10 +2852,21 @@
      "start_time": "2024-06-08T08:17:19.275896Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([1., 4., 9.]) * mvolt2"
+      ]
+     },
+     "execution_count": 104,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 ** 2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1714,10 +2877,21 @@
      "start_time": "2024-06-08T08:17:19.334827Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "20. * mvolt2"
+      ]
+     },
+     "execution_count": 105,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 @ q2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1728,11 +2902,23 @@
      "start_time": "2024-06-08T08:17:19.456793Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "(ArrayImpl([ 0.,  4.,  8., 12., 16.]) * volt,\n",
+       " ArrayImpl([0., 0., 0., 0., 1.]) * volt)"
+      ]
+     },
+     "execution_count": 106,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q1 = bu.Quantity(jnp.arange(5, dtype=jnp.int32), unit=bu.mV.unit)\n",
     "q1 << 2, q1 >> 2"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1743,6 +2929,17 @@
      "start_time": "2024-06-08T08:17:19.514038Z"
     }
    },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "round(80.23456, 2) :  80.23 mV\n",
+      "round(100.000056, 3) :  100. mV\n",
+      "round(-100.000056, 3) :  -100. mV\n"
+     ]
+    }
+   ],
    "source": [
     "q1 = 80.23456 * bu.mV\n",
     "q2 = 100.000056 * bu.mV\n",
@@ -1750,8 +2947,7 @@
     "print(\"round(80.23456, 2) : \", q1.round(5))\n",
     "print(\"round(100.000056, 3) : \", q2.round(6))\n",
     "print(\"round(-100.000056, 3) : \", q3.round(6))"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1770,11 +2966,23 @@
      "start_time": "2024-06-08T08:17:19.670457Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[1., 2.],\n",
+       "           [3., 4.]]) * mvolt"
+      ]
+     },
+     "execution_count": 108,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q = [[1, 2], [3, 4]] * bu.mV\n",
     "q"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1785,10 +2993,21 @@
      "start_time": "2024-06-08T08:17:19.743837Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([1., 2., 3., 4.]) * mvolt"
+      ]
+     },
+     "execution_count": 109,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q.flatten()"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1799,10 +3018,22 @@
      "start_time": "2024-06-08T08:17:19.854522Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[1., 3.],\n",
+       "           [2., 4.]]) * mvolt"
+      ]
+     },
+     "execution_count": 110,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q.swapaxes(0, 1)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1813,10 +3044,21 @@
      "start_time": "2024-06-08T08:17:19.907391Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([1., 3.]) * mvolt"
+      ]
+     },
+     "execution_count": 111,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q.take(jnp.array([0, 2]))"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1827,10 +3069,22 @@
      "start_time": "2024-06-08T08:17:20.023150Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[1., 3.],\n",
+       "           [2., 4.]]) * mvolt"
+      ]
+     },
+     "execution_count": 112,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q.transpose()"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1841,10 +3095,22 @@
      "start_time": "2024-06-08T08:17:20.109833Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[1., 2., 1., 2.],\n",
+       "           [3., 4., 3., 4.]]) * mvolt"
+      ]
+     },
+     "execution_count": 113,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q.tile(2)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1855,10 +3121,22 @@
      "start_time": "2024-06-08T08:17:20.213853Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[[1., 2.],\n",
+       "            [3., 4.]]]) * mvolt"
+      ]
+     },
+     "execution_count": 114,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q.unsqueeze(0)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1869,10 +3147,22 @@
      "start_time": "2024-06-08T08:17:20.279806Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[[1., 2.],\n",
+       "            [3., 4.]]]) * mvolt"
+      ]
+     },
+     "execution_count": 115,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q.expand_dims(0)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1883,11 +3173,23 @@
      "start_time": "2024-06-08T08:17:20.397348Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[[1., 2.],\n",
+       "            [3., 4.]]]) * mvolt"
+      ]
+     },
+     "execution_count": 116,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "expand_as_shape = (1, 2, 2)\n",
     "q.expand_as(jnp.zeros(expand_as_shape).shape)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1898,12 +3200,24 @@
      "start_time": "2024-06-08T08:17:20.454902Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[ 1., 30.],\n",
+       "           [10.,  4.]]) * mvolt"
+      ]
+     },
+     "execution_count": 117,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q_put = [[1, 2], [3, 4]] * bu.mV\n",
     "q_put.put([[1, 0], [0, 1]], [10, 30] * bu.mV)\n",
     "q_put"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1914,11 +3228,23 @@
      "start_time": "2024-06-08T08:17:20.573505Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "ArrayImpl([[1., 2.],\n",
+       "           [3., 4.]]) * mvolt"
+      ]
+     },
+     "execution_count": 118,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q_squeeze = [[1, 2], [3, 4]] * bu.mV\n",
     "q_squeeze.squeeze()"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "code",
@@ -1929,11 +3255,22 @@
      "start_time": "2024-06-08T08:17:20.652077Z"
     }
    },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[ArrayImpl([[1., 2.]]) * mvolt, ArrayImpl([[3., 4.]]) * mvolt]"
+      ]
+     },
+     "execution_count": 119,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "q_spilt = [[1, 2], [3, 4]] * bu.mV\n",
     "q_spilt.split(2)"
-   ],
-   "outputs": []
+   ]
   },
   {
    "cell_type": "markdown",