# JAX Toolbox

> A public CI, Docker images, and optimized examples for JAX on NVIDIA GPUs

## Instructions for AI Agents

- For clean Markdown of any page, append `.md` to the page URL
- For section-specific indexes, append `/llms.txt` to any section URL
- For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/jax-toolbox/_mcp/server

## Products

- [JAX Toolbox](https://docs.nvidia.com/jax-toolbox/llms.txt)
  - [Introduction](https://docs.nvidia.com/jax-toolbox/dev/introduction.md)
  - [JAX on NVIDIA GPU Stack](https://docs.nvidia.com/jax-toolbox/dev/getting-started/jax-on-nvidia-gpu-stack.md)
  - [Frameworks & Supported Models](https://docs.nvidia.com/jax-toolbox/dev/frameworks.md)
  - [Build Pipeline Status](https://docs.nvidia.com/jax-toolbox/dev/build-status.md)
  - [Environment Variables](https://docs.nvidia.com/jax-toolbox/dev/environment-variables.md)