add guide for SGLang-Jax on TPUs

[JamesBrianD]

This is a tutorial for running the sglang-jax project on TPUs.

[google-cla]

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

[JamesBrianD]

@bvandermoon Could you please review this pr?

[bvandermoon]

Hey @JamesBrianD, thanks for reaching out and for your contribution. Could you please provide a little more context around this PR?

[JamesBrianD]

Hello @bvandermoon, thanks for reviewing! Happy to provide context.
SGLang-Jax is a new open-source inference engine that the LMSYS team just announced. It's built entirely on JAX/XLA and designed specifically for TPU inference. It comes with all the production features you'd expect: continuous batching, prefix caching, different parallelism strategies, speculative decoding, and custom TPU kernels.
I know there's already MaxText and vLLM-TPU in the recipes, just thought this could be another option for TPU inference that folks might find useful.
Project: https://github.com/sgl-project/sglang-jax
Blog: https://lmsys.org/blog/2025-10-29-sglang-jax/
Let me know if you'd like me to adjust anything in the PR!

[JamesBrianD]

Hi @bvandermoon, friendly ping on this PR.
Whenever you have time, I’d appreciate a quick review. If you think any changes or clarifications are needed, I’m happy to update the PR accordingly.