Skip to content

1.2.0

Choose a tag to compare

View all tags
@github-actions github-actions released this 28 May 09:50
· 145 commits to main since this release
1.2.0
3471901
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

Installation

Via PyPI

pip install pjrt-plugin-tt==1.2.0 --extra-index-url https://pypi.eng.aws.tenstorrent.com/
pip install vllm-tt==1.2.0 --extra-index-url https://pypi.eng.aws.tenstorrent.com/

Via Docker

docker pull ghcr.io/tenstorrent/tt-xla-slim:1.2.0

What's Changed

Full Changelog: 1.1.0...1.2.0

LLM Performance

Model Token/sec/user Batch Token/sec ttft (ms)
facebook/opt-125m 6.0 1 6.0 175.07
pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface 57.0 32 1824.0 281.39
pytorch_Falcon_3_3B_Base_nlp_causal_lm_huggingface 37.0 32 1184.0 385.3
pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface 40.0 32 1280.0 428.1
pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface 22.0 32 704.0 655.24
pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface 68.0 32 2176.0 248.85
pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface 20.0 32 640.0 638.69
pytorch_Mistral_Ministral_8B_Instruct_nlp_causal_lm_huggingface 12.0 32 384.0 304.81
pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface 24.0 32 768.0 462.4
pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface 24.0 32 768.0 457.77
pytorch_Phi-2_Phi_2_nlp_causal_lm_huggingface 11.0 32 352.0 1002.6
pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface 81.0 32 2592.0 286.17
pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface 39.0 32 1248.0 350.69
pytorch_Qwen 2.5_3B_Instruct_nlp_causal_lm_huggingface 33.0 32 1056.0 531.72
pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface 16.0 32 512.0 759.47
pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface 36.0 32 1152.0 451.46
pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface 30.0 32 960.0 490.97
pytorch_Qwen 3_4B_nlp_causal_lm_huggingface 18.0 32 576.0 683.49
pytorch_Qwen 3_8B_nlp_causal_lm_huggingface 13.0 32 416.0 806.92
tiiuae/Falcon3-1B-Base 32.0 1 32.0 50.67

Non-LLM Performance

Model Batch Sample/sec
pytorch_BERT_emrecan/bert-base-turkish-cased-mean-nli-stsb-tr_nlp_embed_gen_huggingface 8 44.0
pytorch_BGE-M3_Base_nlp_embed_gen_custom 4 9.0
pytorch_EfficientNet_Timm_B0_cv_image_cls_timm 8 332.0
pytorch_MNIST_Cnn_Dropout_cv_image_cls_custom 32 14688.0
pytorch_MobileNetV2_Mobilenet_v2_cv_image_cls_torch_hub 12 1252.0
pytorch_Qwen 3_Embedding_4B_nlp_embed_gen_huggingface 32 46.0
pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface 8 1353.0
pytorch_SegFormer_B0_Finetuned_Ade_512_512_cv_image_seg_huggingface 1 38.0
pytorch_Swin_S_cv_image_cls_torchvision 1 9.0
pytorch_U-Net for Conditional Generation_Base_conditional_generation_huggingface 1 3.0
pytorch_Ultra-Fast Lane Detection v2_TuSimple_ResNet34_Backbone_cv_image_seg_github 1 143.0
pytorch_VGG19-UNet_base_cv_image_seg_custom 1 151.0
pytorch_ViT_Base_cv_image_cls_huggingface 8 237.0
pytorch_VoVNet_Ese_Vovnet19b_Dw.ra_In1k_cv_image_cls_timm 8 713.0

Model coverage

Info: Full list of supported models is available in the assets section.

Model task Model architecture Model variant Model framework Inference Training n150 n300 p150 Single device Data parallel Tensor parallel Model source
conditional generation U-Net for Conditional Generation Base pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls AlexNet Custom 1x2 jax ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
cv image cls DINOv2 Small pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls EfficientNet B0 pytorch ✅ ❌ ✅ ✅ ✅ ✅ ✅ ❌ View Source
cv image cls MNIST Cnn Batchnorm jax ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls MNIST Cnn Dropout jax ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls MNIST Cnn Dropout pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls MNIST Cnn Nodropout pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls MNIST Mlp Custom jax ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls MNIST Mlp Custom jax ❌ ✅ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls MNIST Mlp Custom 1x2 jax ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
cv image cls MobileNetV1 Mobilenet v1 pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls MobileNetV2 Mobilenet v2 pytorch ✅ ❌ ✅ ✅ ✅ ✅ ✅ ❌ View Source
cv image cls ResNet ResNet50 HuggingFace High Resolution pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls SegFormer Mit B0 pytorch ✅ ❌ ✅ ✅ ✅ ✅ ✅ ❌ View Source
cv image cls Swin S pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls VGG HF Vgg19 pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image cls ViT Base pytorch ✅ ❌ ✅ ✅ ✅ ✅ ✅ ❌ View Source
cv image cls VoVNet Ese Vovnet19b Dw.ra In1k pytorch ✅ ❌ ✅ ✅ ✅ ✅ ✅ ❌ View Source
cv image seg MaskFormer Swin-B Swin Base Coco pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image seg Ultra-Fast Lane Detection TuSimple ResNet18 Backbone pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv image seg VGG19-UNet base pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
cv img to img Autoencoder linear pytorch ❌ ✅ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv object det Attention DenseUNet Base pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv object det DETR ResNet50 Backbone pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
cv object det OWL-ViT Base Patch32 pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv object det PointPillars pointpillars pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv object det YOLOP Default pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv object det YOLOS Small Small pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv object det YOLOv4 Base pytorch ✅ ❌ ✅ ✅ ✅ ✅ ✅ ❌ View Source
cv object det YOLOv7 Default pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
cv object det YOLOv9 T pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
cv object det ssd512 ssd512 pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
cv panoptic seg Panoptic Segmentation ResNet50 Backbone 1x COCO pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
mm action prediction OpenVLA-OFT Finetuned Libero 10 pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
mm action prediction pi_0 pi0 base pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
mm image text similarity CLIP Base Patch16 pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
mm image text similarity SigLIP Base Patch16 224 pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
mm visual qa Mistral base pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm ALLaM 7B Instruct pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Command_A_Reasoning command-a-reasoning-08-2025 pytorch ✅ ❌ ❌ ❌ ❌ ❌ ❌ ✅ View Source
nlp causal lm Falcon 3 10B Base pytorch ✅ ❌ ❌ ✅ ✅ ✅ ❌ ✅ View Source
nlp causal lm Falcon 3 1B Base pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Falcon 3 3B Base pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Falcon 3 7B Base pytorch ✅ ❌ ❌ ✅ ✅ ✅ ❌ ✅ View Source
nlp causal lm GPT-2 Base jax ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm GPT-2 Xl jax ❌ ✅ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm GPT-OSS 20B pytorch ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
nlp causal lm Gemma 1.1 2B IT pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Gemma 1.1 7B IT pytorch ✅ ❌ ❌ ✅ ✅ ✅ ❌ ✅ View Source
nlp causal lm Gemma 2 27B IT pytorch ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
nlp causal lm Gemma 2 2B IT pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Gemma 2 9B IT pytorch ✅ ❌ ❌ ✅ ✅ ✅ ❌ ✅ View Source
nlp causal lm Llama 3.1 70B pytorch ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
nlp causal lm Llama 3.1 70B Instruct pytorch ✅ ❌ ❌ ❌ ❌ ❌ ❌ ✅ View Source
nlp causal lm Llama 3.1 8B Instruct pytorch ✅ ❌ ❌ ✅ ✅ ✅ ❌ ✅ View Source
nlp causal lm Llama 3.2 1B pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Llama 3.2 3B pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Llama 3.3 70B Instruct pytorch ✅ ❌ ❌ ❌ ❌ ❌ ❌ ✅ View Source
nlp causal lm Mistral 7B INSTRUCT v03 pytorch ✅ ❌ ❌ ✅ ✅ ✅ ❌ ✅ View Source
nlp causal lm Mistral Devstral Small 2505 pytorch ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
nlp causal lm Mistral Large INSTRUCT 2411 pytorch ✅ ❌ ❌ ❌ ❌ ❌ ❌ ✅ View Source
nlp causal lm Mistral Magistral Small 2506 pytorch ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
nlp causal lm Mistral Ministral 8B Instruct pytorch ✅ ❌ ❌ ✅ ✅ ✅ ❌ ✅ View Source
nlp causal lm Mistral Nemo INSTRUCT 2407 pytorch ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
nlp causal lm Mistral Small 24B INSTRUCT 2501 pytorch ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
nlp causal lm Phi-1 Phi 1 jax ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Phi-1 Phi 1 pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Phi-1.5 Phi 1 5 jax ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Phi-1.5 Phi 1 5 pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Phi-2 Phi 2 pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Phi-3 Mini 128K Instruct pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Phi-3 Mini 4K Instruct pytorch ✅ ❌ ❌ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Phi-3 Mini Instruct pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Phi-4 Phi 4 pytorch ✅ ❌ ❌ ✅ ✅ ✅ ❌ ✅ View Source
nlp causal lm Qwen 2 Qwq 32B pytorch ✅ ❌ ❌ ✅ ❌ ❌ ❌ ✅ View Source
nlp causal lm Qwen 2.5 0.5B jax ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Qwen 2.5 0.5B Instruct jax ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Qwen 2.5 0.5B Instruct pytorch ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source
nlp causal lm Qwen 2.5 1.5B Instruct jax ✅ ❌ ✅ ❌ ✅ ✅ ❌ ❌ View Source

Contributors

AleksKnezevic, kmabeeTT, and 35 other contributors
Assets 5
Loading