Fix typo

keras-team · Jul 20, 2024 · 3e3467d · 3e3467d
1 parent fdfc41b
commit 3e3467d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/guides/ipynb/writing_a_custom_training_loop_in_jax.ipynb b/guides/ipynb/writing_a_custom_training_loop_in_jax.ipynb
@@ -224,7 +224,7 @@
     "```\n",
     "\n",
     "Once you have such a function, you can get the gradient function by\n",
-    "specifying `hax_aux` in `value_and_grad`: it tells JAX that the loss\n",
+    "specifying `has_aux` in `value_and_grad`: it tells JAX that the loss\n",
     "computation function returns more outputs than just the loss. Note that the loss\n",
     "should always be the first output.\n",
     "\n",

diff --git a/guides/md/writing_a_custom_training_loop_in_jax.md b/guides/md/writing_a_custom_training_loop_in_jax.md
@@ -160,7 +160,7 @@ def compute_loss_and_updates(trainable_variables, non_trainable_variables, x, y)
 ```
 
 Once you have such a function, you can get the gradient function by
-specifying `hax_aux` in `value_and_grad`: it tells JAX that the loss
+specifying `has_aux` in `value_and_grad`: it tells JAX that the loss
 computation function returns more outputs than just the loss. Note that the loss
 should always be the first output.