tt-xla

official

Python · Apache-2.0 · 73⭐ ·

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

Links

Releases

LATEST 1.3.0 2026-06-29T11:09:10Z Release notes ↗

5 previous releases

1.4.0.dev20260720022630 2026-07-20T03:02:46Z

1.4.0.dev20260719001637 2026-07-19T01:04:33Z

1.4.0.dev20260718001745 2026-07-18T00:51:56Z

1.4.0.dev20260717001916 2026-07-17T00:54:10Z

1.4.0.dev20260716001804 2026-07-16T00:53:44Z

xla pjrt jax pytorch

Works on

wormhole blackhole