Back to Blog
AI & ML 13 min read

I Rebuilt Karpathy's NanoChat in JAX. Here's What XLA Gets Right and What It Gets Dead Wrong.

Omotayo Aina

Omotayo Aina

May 1, 2026

I Rebuilt Karpathy's NanoChat in JAX. Here's What XLA Gets Right and What It Gets Dead Wrong.

AI GDE TPU Sprint 2026 ยท Google TPU Research Cloud Quick summary: We ported Andrej Karpathy's...

Originally published on Dev.to: View original article →

Share this article