diff --git a/examples/meta_learning.ipynb b/examples/meta_learning.ipynb index 74bba498..e30e4292 100644 --- a/examples/meta_learning.ipynb +++ b/examples/meta_learning.ipynb @@ -194,7 +194,6 @@ }, "outputs": [], "source": [ - "@jax.jit\n", "def outer_loss(eta, theta, state, samples):\n", " state.hyperparams['learning_rate'] = jax.nn.sigmoid(eta)\n", "\n",