Remove useless inner jit

PiperOrigin-RevId: 577316748
google-deepmind · Apr 30, 2024 · b78d9cd · b78d9cd
1 parent 8a3ee74
commit b78d9cd
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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",