tenstorrenttenstorrentOfficialยท4 Agent Skills Included

tt-xla

Run PyTorch and JAX models on Tenstorrent hardware

Compiles and runs PyTorch and JAX models on Tenstorrent AI accelerators through a PJRT backend and TT-MLIR compiler. Diagnoses nightly CI failures, benchmark regressions, graph breaks, dtype mismatches, and missing op fusions without manual log digging. Designs multi-chip sharding strategies and fixes model test failures so hardware bring-up moves faster.
npx skills add tenstorrent/tt-xla --all -g -y
Available:

Explains the TT-XLA architecture, build system, environment variables, and test commands so the agent can build, test, and debug the PJRT plugin correctly.

All Skills in This Repository (4)

Pure Emerald Level Indicators

Frequently Asked Questions

FAQPage Schema
How to install tt-xla?โ–ผ

Run `npx skills add tenstorrent/tt-xla --all -g -y` in your terminal to install all skills in this suite globally.

What does tt-xla do?โ–ผ

It connects PyTorch/XLA and JAX to Tenstorrent AI hardware through a PJRT plugin, compiling models via TT-MLIR for single-chip and multi-chip execution.

How to analyze failing nightly CI runs?โ–ผ

Use the analyze-nightly or ci-benchmark-analyzer skills with a GitHub Actions run ID to get a grouped failure report with root causes and performance comparisons.

How to shard a model across multiple Tenstorrent chips?โ–ผ

The sharding-model-analysis skill maps your model architecture, picks a CCL-minimizing tensor or sequence parallelism strategy, and implements it with Shardy annotations.

Can it fix model test failures automatically?โ–ผ

Yes. Dedicated triage skills diagnose bfloat16 dtype errors, missing training inputs, and forward-output unpacking failures, then apply minimal loader fixes and update test configs.

Related Repositories in Software Engineering

View All in Software Engineeringโ†’