[docs] Add customization section

docs/gpu-kernels.md

@@ -0,0 +1,11 @@
+---
+title: Customize JAX with GPU kernels
+subtitle: Writing custom kernels for JAX on GPUs.
+slug: gpu-kernels
+---
+
+Developers need customization of the JAX stack to customize their models and to optimize beyond what's natively available in JAX/XLA stack.
+
+NVIDIA offers a suite of tools, libraries and kernel DSLs, helping customization.
+
+- [**Writing High-Performance CuTe DSL kernels in JAX**](https://docs.jax.dev/en/latest/notebooks/cute_dsl_jax.html)

docs/index.yml

@@ -29,11 +29,6 @@ navigation:
       - page: Environment Variables
         path: getting-started/environment-variables.md
 
-  # ==================== Documentation ====================
-  - section: Documentation
-    contents:
-      - page: Guides
-        path: documentation/guides.md
   # ==================== Frameworks ====================
   - section: Frameworks
     contents:
@@ -74,11 +69,11 @@ navigation:
       - page: JAX-vLLM Rollout Offloading Bridge
         path: inference/jax-inference-offloading.md
 
-  # ==================== Tools ====================
-  - section: Tools
+  # ==================== Customization ====================
+  - section: Customization
     contents:
-      - page: Triage Tool
-        path: triage-tool.md
+      - page: GPU Kernels
+        path: gpu-kernels.md
 
   # ==================== Reference ====================
   - section: Reference