tt-xla

official

Python · Apache-2.0 · 74⭐ ·

PJRT device plugin for Tenstorrent hardware. Enables JAX, PyTorch/XLA, and other XLA-based frameworks to target TT accelerators.

Links

Releases

LATEST 1.4.0 2026-07-30T11:06:22Z Release notes ↗

5 previous releases

1.5.0.dev20260831000501 2026-08-31T00:34:47Z

1.5.0.dev20260830000121 2026-08-30T00:31:43Z

1.5.0.dev20260829000418 2026-08-29T00:47:57Z

1.5.0.dev20260828023103 2026-08-28T03:00:53Z

1.5.0.dev20260827010900 2026-08-27T01:39:19Z

xla pjrt jax pytorch

Works on

wormhole blackhole