19:00
2026-07-25
hiraditya.github.io
machine-learning
XLA Up Close: What It Optimizes, and What It Won't
XLA, the compiler behind JAX, TensorFlow, and PyTorch/XLA, optimizes array programs by freezing shapes, statically allocating buffers, and fusing operations against a global cost model, which makes itβ¦