在线训练AI模型时,往往需要不同的编译器和硬件深度协作,流程复杂且效率不高。 TT-XLA 是 Tenstorrent 推出的开源 AI 编译器项目,基于 PJRT 接口无缝连接 JAX 框架和 Tenstorrent 的高性能硬件。它支持通过 jit 编译直接导入 JAX 模型,生成 StableHLO(SHLO)图,再交给 tt-mlir 编译器进行优化执行。 核心亮点: - 基于 PJRT,兼容 JAX 及未来更多框架; - 支持 StableHLO(SHLO)图形转换,提升模型编译效率; - 与 Tenstorrent 硬件紧密集成,发挥加速器最大性能; - 开源透明,适合研究和工业级应用。 TT-XLA主要面向 AI 编译、硬件加速和框架集成,助力深度学习推理和训练的高效实现,是 Tenstorrent AI 生态的重要组成部分,适合开发者和研究者快速部署高效 AI 模型。