Fix heading in JAX_for_LLM_pretraining tutorial (#123)

* Fix heading in JAX_for_LLM_pretraining tutorial * Fix heading in JAX_for_LLM_pretraining.ipynb
jax-ml · Dec 5, 2024 · e1022e1 · e1022e1
1 parent 67a7c94
commit e1022e1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/JAX_for_LLM_pretraining.ipynb b/docs/source/JAX_for_LLM_pretraining.ipynb
@@ -1013,7 +1013,7 @@
    "id": "3813cbf2",
    "metadata": {},
    "source": [
-    "# Profiling for Hyperparameter Tuning"
+    "## Profiling for Hyperparameter Tuning"
    ]
   },
   {

diff --git a/docs/source/JAX_for_LLM_pretraining.md b/docs/source/JAX_for_LLM_pretraining.md
@@ -462,7 +462,7 @@ checkpointer.save('/content/save', state)
 !ls /content/save/
 ```
 
-# Profiling for Hyperparameter Tuning
+## Profiling for Hyperparameter Tuning
 
 ```{code-cell}
 !pip install -Uq tensorboard-plugin-profile tensorflow tensorboard