The TT-Blacksmith project contains optimized training recipes for a variety of machine learning models on Tenstorrent hardware, powered by the TT-Forge compiler stack. Showcasing the compiler's flexibility, it enables the use of popular AI frameworks like PyTorch and JAX for training workflows.
For more information on the project, see our documentation.
TT-Blacksmith shows demonstrations of models using the TT-Forge compiler stack. The repo contains a collection of optimized training examples that show developers how to train machine learning models on Tenstorrent hardware using popular frameworks like PyTorch and JAX via thet TT-Forge compiler.
- Demonstrations: Practical examples and workflows showcasing how to train various ML models on Tenstorrent hardware.
This repo is a part of Tenstorrent’s bounty program. If you are interested in helping to improve tt-forge, please make sure to read the Tenstorrent Bounty Program Terms and Conditions before heading to the issues tab. Look for the issues that are tagged with both “bounty” and difficulty level!