a

Author: lockwo

examples/latent_sde.ipynb

@@ -76,7 +76,9 @@
     "\n",
     "    # sample via SDE solver\n",
     "    def integrate(y0, path_key):\n",
-    "        bm = diffrax.UnsafeBrownianPath(shape=(3,), key=path_key)\n",
+    "        bm = diffrax.UnsafeBrownianPath(\n",
+    "            shape=(3,), key=path_key, levy_area=diffrax.BrownianIncrement\n",
+    "        )\n",
     "        lorenz_sde = diffrax.MultiTerm(\n",
     "            diffrax.ODETerm(drift), diffrax.ControlTerm(diffusion, bm)\n",
     "        )\n",