Added ViT training example

Author: vfdev-5

docs_nnx/examples/index.rst

@@ -17,6 +17,7 @@ Example notebooks guide you through applying Flax models to a variety of differe
    ./gemma
    ./digits_diffusion_model
    ./minigpt
+   ./vit_training
 
 
 Example Projects