Skip to content

Commit 348d585

Browse files
committed
Add customization section
1 parent 648a961 commit 348d585

3 files changed

Lines changed: 11 additions & 23 deletions

File tree

docs/documentation/guides.md

Lines changed: 0 additions & 14 deletions
This file was deleted.

docs/gpu-kernels.md

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
---
2+
title: Writing GPU Kernels
3+
subtitle: Writing custom kernels for JAX on GPUs.
4+
slug: guides
5+
---
6+
7+
- [**Writing High-Performance CuTe DSL kernels in JAX**](https://docs.jax.dev/en/latest/notebooks/cute_dsl_jax.html)

docs/index.yml

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -29,11 +29,6 @@ navigation:
2929
- page: Environment Variables
3030
path: getting-started/environment-variables.md
3131

32-
# ==================== Documentation ====================
33-
- section: Documentation
34-
contents:
35-
- page: Guides
36-
path: documentation/guides.md
3732
# ==================== Frameworks ====================
3833
- section: Frameworks
3934
contents:
@@ -74,11 +69,11 @@ navigation:
7469
- page: JAX-vLLM Rollout Offloading Bridge
7570
path: inference/jax-inference-offloading.md
7671

77-
# ==================== Tools ====================
78-
- section: Tools
72+
# ==================== Customization ====================
73+
- section: Customization
7974
contents:
80-
- page: Triage Tool
81-
path: triage-tool.md
75+
- page: GPU Kernels
76+
path: gpu-kernels.md
8277

8378
# ==================== Reference ====================
8479
- section: Reference

0 commit comments

Comments
 (0)