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 layer moving programs be…