fix main HTML link (#210)

Author: jakevdp

docs/source/JAX_Vision_transformer.ipynb

@@ -1262,7 +1262,7 @@
     "- Model checkpointing and exporting with [Orbax](https://orbax.readthedocs.io/en/latest/).\n",
     "- Optimizers and learning rate scheduling with [Optax](https://optax.readthedocs.io/en/latest/).\n",
     "- Freezing model's parameters using trainable parameters filtering with examples: 1) [`flax.nnx.optimizer.Optimizer.update`](https://flax.readthedocs.io/en/latest/api_reference/flax.nnx/training/optimizer.html#flax.nnx.optimizer.Optimizer.update) and 2) [example 2 on `google/flax` GitHub Issues](https://github.com/google/flax/issues/4167#issuecomment-2324245208).\n",
-    "- Other computer vision tutorials using the [JAX AI Stack](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html)."
+    "- Other computer vision tutorials using the [JAX AI Stack](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html)."
    ]
   }
  ],

docs/source/JAX_Vision_transformer.md

@@ -797,4 +797,4 @@ For further reading, check out:
 - Model checkpointing and exporting with [Orbax](https://orbax.readthedocs.io/en/latest/).
 - Optimizers and learning rate scheduling with [Optax](https://optax.readthedocs.io/en/latest/).
 - Freezing model's parameters using trainable parameters filtering with examples: 1) [`flax.nnx.optimizer.Optimizer.update`](https://flax.readthedocs.io/en/latest/api_reference/flax.nnx/training/optimizer.html#flax.nnx.optimizer.Optimizer.update) and 2) [example 2 on `google/flax` GitHub Issues](https://github.com/google/flax/issues/4167#issuecomment-2324245208).
-- Other computer vision tutorials using the [JAX AI Stack](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html).
+- Other computer vision tutorials using the [JAX AI Stack](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html).

docs/source/JAX_basic_text_classification.ipynb

@@ -1019,7 +1019,7 @@
     "In this tutorial we implemented from scratch a simple convolutional neural network and trained it on a text classification dataset. Trained model shows 87% classification accuracy due to its convolutional nature. Next steps to improve the metrics could be to change the model to a transformer-based architecture.\n",
     "\n",
     "- Model checkpointing and exporting using [Orbax](https://orbax.readthedocs.io/en/latest/).\n",
-    "- Other NLP tutorials in [jax-ai-stack](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html)."
+    "- Other NLP tutorials in [jax-ai-stack](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html)."
    ]
   }
  ],

docs/source/JAX_basic_text_classification.md

@@ -476,4 +476,4 @@ print(f"- Predicted review sentiment: {'positive' if pred_label == 0 else 'negat
 In this tutorial we implemented from scratch a simple convolutional neural network and trained it on a text classification dataset. Trained model shows 87% classification accuracy due to its convolutional nature. Next steps to improve the metrics could be to change the model to a transformer-based architecture.
 
 - Model checkpointing and exporting using [Orbax](https://orbax.readthedocs.io/en/latest/).
-- Other NLP tutorials in [jax-ai-stack](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html).
+- Other NLP tutorials in [jax-ai-stack](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html).

docs/source/JAX_for_PyTorch_users.ipynb

@@ -1464,7 +1464,7 @@
     "\n",
     "\n",
     "Other AI/ML tutorials to check out:\n",
-    "- [JAX AI Stack tutorials](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html)\n",
+    "- [JAX AI Stack tutorials](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html)\n",
     "\n",
     "## Just-In-Time (JIT) compilation in JAX\n",
     "\n",

docs/source/JAX_for_PyTorch_users.md

@@ -809,7 +809,7 @@ More details about Flax NNX API, how to save and load the model's state and abou
 
 
 Other AI/ML tutorials to check out:
-- [JAX AI Stack tutorials](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html)
+- [JAX AI Stack tutorials](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html)
 
 ## Just-In-Time (JIT) compilation in JAX
 

docs/source/JAX_image_captioning.ipynb

@@ -2309,7 +2309,7 @@
     "In this tutorial we implemented and trained a transformer-based model for image captioning task. We used a pretrained frozen Vision Transformer encoder and trained a small decoder to predict the next token. Observed generation capabilities of the trained model are not great. Next steps could be (1) to use larger decoder, (2) to unfreeze few top encoder layers, (3) try other decoder architectures.\n",
     "\n",
     "- Freezing model's parameters using trainable parameters filtering: [example 1](https://flax.readthedocs.io/en/latest/api_reference/flax.nnx/training/optimizer.html#flax.nnx.optimizer.Optimizer.update) and [example 2](https://github.com/google/flax/issues/4167#issuecomment-2324245208).\n",
-    "- Other Computer Vision tutorials in [jax-ai-stack](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html).\n",
+    "- Other Computer Vision tutorials in [jax-ai-stack](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html).\n",
     "- [LLM pretraining for text generation](https://jax-ai-stack.readthedocs.io/en/latest/JAX_for_LLM_pretraining.html)."
    ]
   }

docs/source/JAX_image_captioning.md

@@ -1131,5 +1131,5 @@ for i, (prefix, pil_image) in enumerate([
 In this tutorial we implemented and trained a transformer-based model for image captioning task. We used a pretrained frozen Vision Transformer encoder and trained a small decoder to predict the next token. Observed generation capabilities of the trained model are not great. Next steps could be (1) to use larger decoder, (2) to unfreeze few top encoder layers, (3) try other decoder architectures.
 
 - Freezing model's parameters using trainable parameters filtering: [example 1](https://flax.readthedocs.io/en/latest/api_reference/flax.nnx/training/optimizer.html#flax.nnx.optimizer.Optimizer.update) and [example 2](https://github.com/google/flax/issues/4167#issuecomment-2324245208).
-- Other Computer Vision tutorials in [jax-ai-stack](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html).
+- Other Computer Vision tutorials in [jax-ai-stack](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html).
 - [LLM pretraining for text generation](https://jax-ai-stack.readthedocs.io/en/latest/JAX_for_LLM_pretraining.html).

docs/source/JAX_porting_PyTorch_model.ipynb

@@ -2271,7 +2271,7 @@
     "## Further reading\n",
     "\n",
     "- [Flax documentation: Core Examples](https://flax.readthedocs.io/en/latest/examples/core_examples.html)\n",
-    "- [JAX AI Stack tutorials](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html)"
+    "- [JAX AI Stack tutorials](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html)"
    ]
   }
  ],

docs/source/JAX_porting_PyTorch_model.md

@@ -1630,4 +1630,4 @@ cosine_dist
 ## Further reading
 
 - [Flax documentation: Core Examples](https://flax.readthedocs.io/en/latest/examples/core_examples.html)
-- [JAX AI Stack tutorials](https://jax-ai-stack.readthedocs.io/en/latest/tutorials.html)
+- [JAX AI Stack tutorials](https://jax-ai-stack.readthedocs.io/en/latest/getting_started.html)