XLA 是 Google 主导的开源机器学习编译器，也是 TPU 软件栈的核心，负责接收计算图并进行算子融合、内存布局优化和常量折叠等变换，最终生成 TPU 可执行代码。