Updates to "Customizing fit() with JAX" guide.#2188

hertschuh merged 1 commit intokeras-team:masterkeras-team/keras-io:masterfrom

hertschuh:jax_fithertschuh/keras-io:jax_fitCopy head branch name to clipboard