Posts by Ruturaj Kiran Vaidya
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.