1.2.0.dev20260531003057

Installation

Via PyPI

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

Via Docker

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

What's Changed

• uplift torch_xla by @mstojkovicTT in #4979
• [CI] Add QB2-Blackhole TP benchmarks to nightly by @rpavlovicTT in #4983
• Update vLLM benchmark CODEOWNERS by @vkovacevicTT in #4984
• Add training mode and LoRA backward tests for LLM torch models by @agobeljicTT in #4219
• [Benchmarks] Remove hardcoded arch report parameter by @vkovacevicTT in #4972
• Add Gemma4 e4b, 31b support for vLLM by @sshonTT in #4889

Full Changelog: 1.2.0.dev20260529003234...1.2.0.dev20260531003057

---

LLM Performance

Model	Token/sec/user	Batch	Token/sec	ttft (ms)
Qwen/Qwen2.5-0.5B-Instruct	2.0	1	2.0	581.74
Qwen/Qwen2.5-1.5B-Instruct	1.0	1	1.0	975.33
Qwen/Qwen2.5-3B-Instruct	1.0	1	1.0	1352.24
Qwen/Qwen3-0.6B	2.0	1	2.0	672.12
Qwen/Qwen3-1.7B	1.0	1	1.0	1272.87
facebook/opt-125m	6.0	1	6.0	171.2
meta-llama/Llama-3.2-1B-Instruct	37.0	1	37.0	44.61
microsoft/phi-1	30.0	1	30.0	64.84
microsoft/phi-1_5	30.0	1	30.0	65.08
microsoft/phi-2	17.0	1	17.0	111.39
pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface	39.0	32	1248.0	273.24
pytorch_Falcon_3_3B_Base_nlp_causal_lm_huggingface	26.0	32	832.0	369.86
pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface	33.0	32	1056.0	383.14
pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface	14.0	32	448.0	594.77
pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface	43.0	32	1376.0	236.86
pytorch_Llama_3.2_3B_Instruct_nlp_causal_lm_huggingface	21.0	32	672.0	409.4
pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface	13.0	32	416.0	563.76
pytorch_Mistral_Ministral_8B_Instruct_nlp_causal_lm_huggingface	12.0	32	384.0	252.11
pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface	22.0	32	704.0	412.81
pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface	22.0	32	704.0	443.4
pytorch_Phi-2_Phi_2_nlp_causal_lm_huggingface	10.0	32	320.0	845.36
pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface	57.0	32	1824.0	290.69
pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface	26.0	32	832.0	318.14
pytorch_Qwen 2.5_3B_Instruct_nlp_causal_lm_huggingface	23.0	32	736.0	505.92
pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface	12.0	32	384.0	660.6
pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface	28.0	32	896.0	452.03
pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface	24.0	32	768.0	459.14
pytorch_Qwen 3_4B_nlp_causal_lm_huggingface	14.0	32	448.0	665.97
pytorch_Qwen 3_8B_nlp_causal_lm_huggingface	11.0	32	352.0	719.23
tiiuae/Falcon3-1B-Base	32.0	1	32.0	50.26
tiiuae/Falcon3-3B-Base	12.0	1	12.0	122.33

Non-LLM Performance

Model	Batch	Sample/sec
BAAI/bge-m3	1	39.0
Qwen/Qwen3-Embedding-4B	1	7.0
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	14441.0
pytorch_MobileNetV2_Mobilenet_v2_cv_image_cls_torch_hub	12	1249.0
pytorch_Qwen 3_Embedding_4B_nlp_embed_gen_huggingface	32	47.0
pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface	8	1350.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	5.0
pytorch_Ultra-Fast Lane Detection v2_TuSimple_ResNet34_Backbone_cv_image_seg_github	1	144.0
pytorch_VGG19-UNet_base_cv_image_seg_custom	1	141.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	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
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	Llama	3.2 11B Vision Instruct	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
nlp causal lm	Qwen 2.5	14B Instruct	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source