03:00
2026-07-24
hiraditya.github.io
machine-learning
A Tour of XLA: Where MLIR Lives (and Where It Doesn't)
XLA, the compiler under JAX, TensorFlow, and PyTorch/XLA, uses two intermediate representations: classic HLO (a hand-built C++ IR) and MLIR dialects such as StableHLO and CHLO, with a translation laye…