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.dev20260820000439 2026-08-20T00:35:46Z
1.5.0.dev20260819104117 2026-08-19T11:28:45Z
1.5.0.dev20260819000359 2026-08-19T00:35:41Z
1.5.0.dev20260818002307 2026-08-18T00:54:16Z
1.5.0.dev20260817000312 2026-08-17T00:35:09Z
See all releases on GitHub ↗
xla pjrt jax pytorch
Works on
wormhole blackhole
Related