Update tpu examples for new folder structure

examples/fine-tuning/optimum-tpu/gemma/.dstack.yml

@@ -26,4 +26,4 @@ ide: vscode
 spot_policy: auto
 
 resources:
-  gpu: tpu-v5litepod-8
+  gpu: v5litepod-8

examples/fine-tuning/optimum-tpu/gemma/train.dstack.yml

@@ -21,4 +21,4 @@ ports:
   - 6006
 
 resources:
-  gpu: tpu-v5litepod-8
+  gpu: v5litepod-8

examples/fine-tuning/optimum-tpu/llama31/.dstack.yml

@@ -19,14 +19,13 @@ env:
 #  - cp examples/fine-tuning/optimum-tpu/llama31/config.yaml optimum-tpu/examples/custom/config.yaml
 #  - cd optimum-tpu
 #  - pip install -e . -f https://storage.googleapis.com/libtpu-releases/index.html
-#  - pip install trl peft
 #  - pip install datasets evaluate
-#  - pip install accelerate>=0.21.0
+#  - pip install accelerate -U
 
 ide: vscode
 
 # Use either spot or on-demand instances
 spot_policy: auto
 
 resources:
-  gpu: tpu-v5litepod-8
+  gpu: v5litepod-8

examples/fine-tuning/optimum-tpu/llama31/train.dstack.yml

@@ -23,4 +23,4 @@ ports:
   - 6006
 
 resources:
-  gpu: tpu-v5litepod-8
+  gpu: v5litepod-8

mkdocs.yml

@@ -224,13 +224,15 @@ nav:
               - Python API: docs/reference/api/python/index.md
               - REST API: docs/reference/api/rest/index.md
   - Examples:
+<<<<<<< HEAD
     - examples.md
     - Fine-tuning:
         - Axolotl: examples/fine-tuning/axolotl/index.md
         - Alignment Handbook: examples/fine-tuning/alignment-handbook/index.md
         - TRL: examples/fine-tuning/trl/index.md
     - Accelerators:
         - AMD: examples/accelerators/amd/index.md
+        - TPU: docs/examples/accelerators/tpu/index.md
     - LLMs:
         - Llama 3.1: examples/llms/llama31/index.md
   - Blog: