Open in tt-awesome →

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
📦 Repo 📖 JAX and PyTorch/XLA on Tenstorrent 🌐 Website
Releases
LATEST 1.4.0 2026-07-30T11:06:22Z Release notes ↗
5 previous releases
1.5.0.dev20260814001246 2026-08-14T00:42:06Z
1.5.0.dev20260812001218 2026-08-12T00:43:01Z
1.5.0.dev20260811000841 2026-08-11T00:38:18Z
1.5.0.dev20260810000801 2026-08-10T00:37:09Z
1.5.0.dev20260809000535 2026-08-09T00:34:42Z
See all releases on GitHub ↗
xla pjrt jax pytorch
Works on
wormhole blackhole
Related