A Tour of XLA: The Compiler Beneath JAX, TensorFlow, and PyTorch
If you train or serve models, XLA is probably compiling them, whether or not you have ever named it. It is the compiler behind JAX, behind TensorFlow, and — through PyTorch/XLA — behind much of PyT...