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