Search results for “JAX”

Matching articles

CODINGJuly 24, 2026

Run Ray on TPU, Part 2: Ray AI libraries

Google Developers Blog

CODINGJuly 21, 2026

Scaling Agentic RL: High-Throughput Agentic Training with Tunix

Google Developers Blog

CODINGJuly 14, 2026

Systems Engineering Playbook: Optimizing Qwen 3.5-397B MoE on Ironwood (TPU7x)

Google Developers Blog

CODINGApril 27, 2026

Fine-Tune Any HuggingFace Model like Gemma on TPUs with TorchAX