Skip to content

Commit a029c40

Browse files
committed
Add guides section
1 parent e0aeef0 commit a029c40

3 files changed

Lines changed: 21 additions & 2 deletions

File tree

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,12 @@
33
JAX Toolbox
44
===========================
55

6-
**JAX-Toolbox** is NVIDIA's home for [JAX and XLA on GPUs](https://docs.nvidia.com/jax-toolbox/getting-started/jax-on-nvidia-gpu-stack), including the latest updates, containers and educational resources.
6+
**JAX Toolbox** is NVIDIA's home for [**JAX and XLA on GPUs**](https://docs.nvidia.com/jax-toolbox/getting-started/jax-on-nvidia-gpu-stack), including the latest updates, containers and educational resources.
77

88
[![License Apache 2.0](https://badgen.net/badge/license/apache2.0/blue)](https://github.com/NVIDIA/JAX-Toolbox/blob/main/LICENSE.md)
99
[![Build](https://badgen.net/badge/build/check-status/blue)](#build-pipeline-status)
1010

11-
[**Latest news**](https://github.com/NVIDIA/JAX-Toolbox/#latest-news)   |   [**Tech blogs**](https://github.com/NVIDIA/JAX-Toolbox/#tech-blogs)   |   [**Documentation**](docs.nvidia.com/jax-toolbox/introduction)   |   [**Resources**](https://github.com/NVIDIA/JAX-Toolbox/#resources)   |   [**Container images**](https://github.com/NVIDIA/JAX-Toolbox/#container-images)
11+
[**Latest news**](https://github.com/NVIDIA/JAX-Toolbox/#latest-news)   |   [**Tech blogs**](https://github.com/NVIDIA/JAX-Toolbox/#tech-blogs)   |   [**Documentation**](https://docs.nvidia.com/jax-toolbox/introduction)   |   [**Resources**](https://github.com/NVIDIA/JAX-Toolbox/#resources)   |   [**Container images**](https://github.com/NVIDIA/JAX-Toolbox/#container-images)
1212

1313
---
1414
<div align="left">

docs/documentation/guides.md

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
---
2+
title: Guides
3+
subtitle: Practical guides for profiling, optimizing, scaling and third-party tool integrations in JAX.
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)
8+
- [**Tips for High-Performance LLMs with JAX on GPUs**](https://github.com/NVIDIA/JAX-Toolbox/blob/main/docs/GPU_performance.md)
9+
- [**AutoPGLE on GPU workflows**](https://github.com/NVIDIA/JAX-Toolbox/blob/main/docs/PGLE.md)
10+
- [**Profiling JAX programs on GPU**](https://github.com/NVIDIA/JAX-Toolbox/blob/main/docs/profiling.md)
11+
- [**`nsys-jax`**](https://github.com/NVIDIA/JAX-Toolbox/blob/main/docs/nsys-jax.md)
12+
- [**Resilient Training with Ray and JAX**](https://github.com/NVIDIA/JAX-Toolbox/blob/main/docs/resiliency/ray_resilient_jax.md)
13+
- [**User Guide on Native XLA-FP8**](https://github.com/NVIDIA/JAX-Toolbox/blob/main/docs/NATIVE_FP8.md)
14+

docs/index.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,11 @@ 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
3237
# ==================== Frameworks ====================
3338
- section: Frameworks
3439
contents:

0 commit comments

Comments
 (0)