decreased batch size to allow execution in notebook

examples/jaxsim_as_physics_engine.ipynb

@@ -173,7 +173,7 @@
     "key = jax.random.PRNGKey(seed=0)\n",
     "\n",
     "# Initialize the simulated time.\n",
-    "T = jnp.arange(start=0, stop=1.0, step=model.time_step)\n",
+    "T = jnp.arange(start=0, stop=0.3, step=model.time_step)\n",
     "\n",
     "# Simulate\n",
     "for _t in T:\n",
@@ -236,7 +236,7 @@
     "# Then we have to create the vector of initial state\n",
     "\n",
     "# Split subkeys for sampling random initial data.\n",
-    "batch_size = 10\n",
+    "batch_size = 5\n",
     "key, *subkeys = jax.random.split(key=key, num=batch_size + 1)\n",
     "\n",
     "# Create the batched data.\n",