From b48159e103712fcbf1e2d473a7ab1654346fa15e Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Thu, 22 Feb 2024 11:44:28 -0800
Subject: [PATCH] [DOCS]

---
 mkdocs.yml | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/mkdocs.yml b/mkdocs.yml
index 35697146..a49a8c6b 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -272,9 +272,10 @@ nav:
             - DecoupledLionW: "zeta/training/optimizers/decoupled_lion.md"
             - SophiaG: "zeta/training/optimizers/sophia.md"
       - zeta.training:
-          - fsdp: "zeta/training/fsdp.md"
-          - ParallelWrapper: "zeta/training/parallel_wrapper.md"
-          - train: "zeta/training/train.md"
+          - Training:
+            - fsdp: "zeta/training/fsdp.md"
+            - ParallelWrapper: "zeta/training/parallel_wrapper.md"
+            - train: "zeta/training/train.md"
       - zeta.models:
           - Language and MultiModal:
             - vit: "zeta/models/vit.md"
@@ -301,11 +302,13 @@ nav:
             - encoder: "encoder.md"
             - encoderdecoder: "encoderdecoder.md"
       - zeta.quant:
-          - QUIK: "zeta/quant/quik.md"
-          - BitLinear: "zeta/quant/bitlinear.md"
-          - niva: "zeta/quant/niva.md"
+          - Quantization Algorithms:
+            - QUIK: "zeta/quant/quik.md"
+            - BitLinear: "zeta/quant/bitlinear.md"
+            - niva: "zeta/quant/niva.md"
       - zeta.rl:
-          - DPO: "zeta/rl/dpo.md"
+          - Reinforcement Learning:
+            - DPO: "zeta/rl/dpo.md"
   - Examples:
     - Overview: "examples/index.md"
     - PytorchCS: "examples/torch_cs.md"