Posts by Jehandad Khan
JAX-AITER: Bringing AMD’s Optimized AI Kernels to JAX on ROCm™
- 24 February 2026
If you’re building large models in JAX on AMD GPUs, you want fast, reliable kernels without spending weeks tuning them yourself. That’s exactly the need that led us to create JAX-AITER.
ROCm MaxText Testing — Decoupled (Offline) and Cloud-Integrated Modes
- 06 January 2026
In this blog, you will learn how to run MaxText unit tests on AMD ROCm GPUs in two complementary modes: offline (decoupled) and fully cloud-integrated. By the end, you will know when to use each mode, how to interpret the results, and how to fold them into your CI and debugging workflows.
ROCm Fork of MaxText: Structure and Strategy
- 06 January 2026
In this blog you will explore how the ROCm fork of MaxText is structured and how that structure supports ROCm and fully offline, decoupled workflows across platforms.