1.5.0.dev20260817000312

Installation

Via PyPI

pip install tt-forge==1.5.0.dev20260817000312 --extra-index-url https://pypi.eng.aws.tenstorrent.com/

Via Docker

docker pull ghcr.io/tenstorrent/tt-forge-slim:1.5.0.dev20260817000312

Dependency commits

tt-xla commit: 1c9f93e45db98d81814e850a748f35a53db2a0ab
tt-mlir commit: 71046369d603b97fd6a8dd8b947ca8588ac2a74f
tt-metal commit: 5beed318d0f0d1c6212e605947fb0be80c9e0a1d

What's Changed

• Uplift third_party/tt_forge_models to 464c104944cb42b25ac83e02f03a5a238572eda2 2026-08-15 by @vvukomanTT in #1025

Full Changelog: 1.5.0.dev20260815000421...1.5.0.dev20260817000312

---

LLM Performance

Model	Token/sec/user	Batch	Token/sec	ttft (ms)	Hardware
pytorch_Falcon_3_10B_Base_nlp_causal_lm_huggingface	41.0	32	1312.0	855.39	p150
pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface	54.0	32	1728.0	692.4	n150
pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface	104.0	32	3328.0	300.18	p150
pytorch_Falcon_3_3B_Base_nlp_causal_lm_huggingface	35.0	32	1120.0	858.69	n150
pytorch_Falcon_3_7B_Base_nlp_causal_lm_huggingface	18.0	32	576.0	1187.5	n150
pytorch_Falcon_3_7B_Base_nlp_causal_lm_huggingface	56.0	32	1792.0	612.85	p150
pytorch_GPT-OSS_20B_nlp_causal_lm_huggingface	8.0	64	512.0	6915.9	n150
pytorch_GPT-OSS_20B_nlp_causal_lm_huggingface	21.0	1	21.0	300.65	p150
pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface	38.0	32	1216.0	638.96	n150
pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface	76.0	32	2432.0	240.27	p150
pytorch_Llama_3.1_70B_Instruct_nlp_causal_lm_huggingface	4.0	32	128.0	11686.24	n150
pytorch_Llama_3.1_70B_Instruct_nlp_causal_lm_huggingface	12.0	32	384.0	1797.85	p150
pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface	21.0	32	672.0	1293.75	n150
pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface	51.0	32	1632.0	653.36	p150
pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface	64.0	32	2048.0	568.65	n150
pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface	125.0	32	4000.0	239.98	p150
pytorch_Llama_3.2_3B_Instruct_nlp_causal_lm_huggingface	30.0	32	960.0	611.32	n150
pytorch_Llama_3.2_3B_Instruct_nlp_causal_lm_huggingface	53.0	32	1696.0	282.61	p150
pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface	20.0	32	640.0	1262.1	n150
pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface	34.0	32	1088.0	572.87	p150
pytorch_Mistral_Ministral_8B_Instruct_nlp_causal_lm_huggingface	15.0	32	480.0	571.58	n150
pytorch_Mistral_Ministral_8B_Instruct_nlp_causal_lm_huggingface	30.0	32	960.0	223.94	p150
pytorch_Mistral_Small_24B_INSTRUCT_2501_nlp_causal_lm_huggingface	29.0	32	928.0	888.03	p150
pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface	20.0	32	640.0	647.16	n150
pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface	37.0	32	1184.0	317.89	p150
pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface	20.0	32	640.0	649.08	n150
pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface	37.0	32	1184.0	316.66	p150
pytorch_Phi-2_Phi_2_nlp_causal_lm_huggingface	8.0	32	256.0	1423.43	n150
pytorch_Qwen 2.5 Coder_32B_Instruct_nlp_causal_lm_huggingface	18.0	32	576.0	1472.83	p150
pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface	65.0	32	2080.0	420.67	n150
pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface	126.0	32	4032.0	171.0	p150
pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface	35.0	32	1120.0	469.63	n150
pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface	62.0	32	1984.0	207.41	p150
pytorch_Qwen 2.5_3B_Instruct_nlp_causal_lm_huggingface	29.0	32	928.0	693.28	n150
pytorch_Qwen 2.5_3B_Instruct_nlp_causal_lm_huggingface	57.0	32	1824.0	284.55	p150
pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface	16.0	32	512.0	824.15	n150
pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface	28.0	32	896.0	352.83	p150
pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface	49.0	32	1568.0	1175.78	n150
pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface	96.0	32	3072.0	545.47	p150
pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface	36.0	32	1152.0	735.59	n150
pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface	65.0	32	2080.0	337.86	p150
pytorch_Qwen 3_32B_nlp_causal_lm_huggingface	11.0	32	352.0	1902.67	p150
pytorch_Qwen 3_4B_nlp_causal_lm_huggingface	22.0	32	704.0	995.05	n150
pytorch_Qwen 3_4B_nlp_causal_lm_huggingface	40.0	32	1280.0	454.96	p150
pytorch_Qwen 3_8B_nlp_causal_lm_huggingface	17.0	32	544.0	1591.76	n150
pytorch_Qwen 3_8B_nlp_causal_lm_huggingface	30.0	32	960.0	751.06	p150

Non-LLM Performance

Model	Batch	Sample/sec	Hardware
Wan2.2-I2V-A14B-DiT	1	0.0	p150
Wan2.2-I2V-A14B-UMT5-Text-Encoder	1	12.0	p150
Wan2.2-I2V-A14B-VAE-Decoder	1	0.0	p150
Wan2.2-I2V-A14B-VAE-Encoder	1	1.0	p150
flux1-dev	1	0.0	p150
flux2	1	0.0	p150
glm-image	1	0.0	p150
hunyuan-image-2.1	1	0.0	p150
janus-pro-1b	1	0.0	n150
janus-pro-1b	1	0.0	p150
janus-pro-7b	1	0.0	p150
playground-v2.5	1	0.0	n150
playground-v2.5	1	0.0	p150
pytorch_BERT_emrecan/bert-base-turkish-cased-mean-nli-stsb-tr_nlp_embed_gen_huggingface	8	158.0	n150
pytorch_BGE-M3_Base_nlp_embed_gen_custom	4	9.0	n150
pytorch_BGE-M3_Base_nlp_embed_gen_custom	4	17.0	p150
pytorch_EfficientNet_Timm_B0_cv_image_cls_timm	8	346.0	n150
pytorch_EfficientNet_Timm_B0_cv_image_cls_timm	8	794.0	p150
pytorch_MNIST_Cnn_Dropout_cv_image_cls_custom	32	14016.0	n150
pytorch_MNIST_Cnn_Dropout_cv_image_cls_custom	32	30176.0	p150
pytorch_MobileNetV2_Mobilenet_v2_cv_image_cls_torch_hub	12	1170.0	n150
pytorch_MobileNetV2_Mobilenet_v2_cv_image_cls_torch_hub	12	2789.0	p150
pytorch_Qwen 3_Embedding_4B_nlp_embed_gen_huggingface	32	46.0	n150
pytorch_Qwen 3_Embedding_4B_nlp_embed_gen_huggingface	32	96.0	p150
pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface	8	1309.0	n150
pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface	8	2759.0	p150
pytorch_SegFormer_B0_Finetuned_Ade_512_512_cv_image_seg_huggingface	1	37.0	n150
pytorch_SegFormer_B0_Finetuned_Ade_512_512_cv_image_seg_huggingface	1	82.0	p150
pytorch_Swin_S_cv_image_cls_torchvision	1	10.0	n150
pytorch_Swin_S_cv_image_cls_torchvision	1	22.0	p150
pytorch_U-Net for Conditional Generation_Base_conditional_generation_huggingface	1	5.0	n150
pytorch_U-Net for Conditional Generation_Base_conditional_generation_huggingface	1	9.0	p150
pytorch_Ultra-Fast Lane Detection v2_TuSimple_ResNet34_Backbone_cv_image_seg_github	1	137.0	n150
pytorch_Ultra-Fast Lane Detection v2_TuSimple_ResNet34_Backbone_cv_image_seg_github	1	235.0	p150
pytorch_VGG19-UNet_base_cv_image_seg_custom	1	150.0	n150
pytorch_VGG19-UNet_base_cv_image_seg_custom	1	311.0	p150
pytorch_ViT_Base_cv_image_cls_huggingface	8	234.0	n150
pytorch_ViT_Base_cv_image_cls_huggingface	8	563.0	p150
pytorch_VoVNet_Ese_Vovnet19b_Dw.ra_In1k_cv_image_cls_timm	8	743.0	n150
pytorch_VoVNet_Ese_Vovnet19b_Dw.ra_In1k_cv_image_cls_timm	8	1549.0	p150
sdxl-lightning	1	0.0	n150
sdxl-lightning	1	0.0	p150
zimage	1	0.0	p150

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
nlp causal lm	Qwen 2	Qwq 32B	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
cv object det	YOLOv4	Base	pytorch	✅	❌	✅	✅	✅	✅	✅	❌	View Source
nlp causal lm	GPT-2	Xl	jax	❌	✅	✅	❌	✅	✅	❌	❌	View Source
cv image seg	Ultra-Fast Lane Detection	TuSimple ResNet18 Backbone	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm image ttt	FLUX	Vae	pytorch	✅	❌	❌	❌	❌	✅	❌	❌	Source not available
nlp causal lm	Qwen 2.5	1.5B Instruct	jax	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	Falcon	3 10B Base	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
cv object det	YOLOv7	Default	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv object det	EfficientDet	D0	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 3	0 6B	pytorch	✅	❌	✅	✅	✅	✅	❌	✅	View Source
cv img to img	Autoencoder	linear	pytorch	❌	✅	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 3	4B	jax	✅	❌	❌	✅	❌	❌	❌	✅	View Source
cv object det	PointPillars	pointpillars	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Llama	3.1 8B Instruct	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp causal lm	Llama LoRA	3.2 1B	pytorch	❌	✅	✅	❌	✅	✅	❌	❌	View Source
cv image cls	DINOv2	Small	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm tts	xtts_v2	gpt prefill	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Llama	3.2 1B	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-3	Mini Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-1	Phi 1	pytorch	❌	✅	❌	❌	✅	✅	❌	❌	View Source
cv image cls	ResNet	ResNet50 HuggingFace High Resolution	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	MNIST	Mlp Custom	jax	❌	✅	✅	❌	✅	✅	❌	❌	View Source
mm image ttt	FLUX	Transformer	pytorch	✅	❌	❌	❌	❌	✅	❌	❌	Source not available
nlp causal lm	Mistral	Small 24B INSTRUCT 2501	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	Qwen 2.5	0.5B Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-1	Phi 1	jax	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	Phi-3	Mini 4K Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp embed gen	FLUX	ClipTextEncoder	pytorch	✅	❌	❌	❌	❌	✅	❌	❌	Source not available
mm image text similarity	SigLIP	Base Patch16 224	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm tts	xtts_v2	conditioning	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	olmo_3	3 1025 7b	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
cv object det	ssd512	ssd512	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv object det	Attention DenseUNet	Base	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Falcon	3 3B Base	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	MNIST	Cnn Nodropout	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	ALLaM	7B Instruct	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5	32B Instruct	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	Mistral	Devstral Small 2505	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp token cls	BiLSTM-CRF	Default	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Gemma	2 27B IT	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	Phi-4	Phi 4	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp causal lm	SOLAR-10_7B	10 7B Instruct v1.0	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
cv image cls	MNIST	Cnn Batchnorm	jax	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Gemma	1.1 2B IT	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	arcee_ai	arcee Spark	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-1.5	Phi 1 5	jax	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	Falcon	3 7B Base	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
cv object det	OWL-ViT	Base Patch32	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-2	Phi 2	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	Swin	S	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	ViT	Base	pytorch	✅	❌	✅	✅	✅	✅	✅	❌	View Source
cv object det	YOLOv9	T	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	olmo_3	3 7b instruct	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
nlp causal lm	GPT-OSS	20B	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
cv image cls	MNIST	Cnn Dropout	jax	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5 Coder	0.5B	jax	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp embed gen	BERT	emrecan/bert-base-turkish-cased-mean-nli-stsb-tr	pytorch	✅	❌	❌	✅	❌	❌	✅	❌	View Source
mm visual qa	Llama	3.2 11B Vision Instruct	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
nlp causal lm	Gemma	2 2B IT	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv panoptic seg	Panoptic Segmentation	ResNet50 Backbone 1x COCO	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5	14B Instruct	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp embed gen	Qwen 3	Embedding 8B	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv object det	DETR	ResNet50 Backbone	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5	7B Instruct	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp causal lm	Mistral	7B INSTRUCT v03	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
mm tts	xtts_v2	hifigan decoder	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	AlexNet	Custom 1x2	jax	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	Qwen 2.5	3B Instruct	jax	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	Mistral	Nemo INSTRUCT 2407	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
nlp causal lm	Falcon	3 1B Base	pytorch	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	GPT-2	Base	jax	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5	3B Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Llama	3.2 3B	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	MobileNetV2	Mobilenet v2	pytorch	✅	❌	✅	✅	✅	✅	✅	❌	View Source
nlp causal lm	Gemma	2 9B IT	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp causal lm	Qwen 3	4B	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-1.5	Phi 1 5	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp embed gen	Qwen 3	Embedding 4B	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm visual qa	Mistral	base	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp embed gen	FLUX	T5TextEncoder	pytorch	✅	❌	❌	❌	❌	✅	❌	❌	Source not available

---

LLM Performance

Model	Token/sec/user	Batch	Token/sec	ttft (ms)	Hardware
pytorch_Falcon_3_10B_Base_nlp_causal_lm_huggingface	41.0	32	1312.0	880.75	p150
pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface	54.0	32	1728.0	671.86	n150
pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface	101.0	32	3232.0	295.89	p150
pytorch_Falcon_3_3B_Base_nlp_causal_lm_huggingface	35.0	32	1120.0	898.87	n150
pytorch_Falcon_3_3B_Base_nlp_causal_lm_huggingface	64.0	32	2048.0	380.1	p150
pytorch_Falcon_3_7B_Base_nlp_causal_lm_huggingface	18.0	32	576.0	1196.82	n150
pytorch_Falcon_3_7B_Base_nlp_causal_lm_huggingface	35.0	32	1120.0	511.22	p150
pytorch_GPT-OSS_20B_nlp_causal_lm_huggingface	8.0	64	512.0	6916.37	n150
pytorch_GPT-OSS_20B_nlp_causal_lm_huggingface	21.0	1	21.0	300.68	p150
pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface	38.0	32	1216.0	629.85	n150
pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface	76.0	32	2432.0	243.13	p150
pytorch_Llama_3.1_70B_Instruct_nlp_causal_lm_huggingface	7.0	32	224.0	9606.17	n150
pytorch_Llama_3.1_70B_Instruct_nlp_causal_lm_huggingface	12.0	32	384.0	1812.13	p150
pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface	21.0	32	672.0	1313.17	n150
pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface	51.0	32	1632.0	659.73	p150
pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface	64.0	32	2048.0	596.92	n150
pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface	121.0	32	3872.0	233.73	p150
pytorch_Llama_3.2_3B_Instruct_nlp_causal_lm_huggingface	30.0	32	960.0	600.73	n150
pytorch_Llama_3.2_3B_Instruct_nlp_causal_lm_huggingface	55.0	32	1760.0	282.84	p150
pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface	19.0	32	608.0	1267.12	n150
pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface	35.0	32	1120.0	582.82	p150
pytorch_Mistral_Ministral_8B_Instruct_nlp_causal_lm_huggingface	13.0	32	416.0	611.13	n150
pytorch_Mistral_Ministral_8B_Instruct_nlp_causal_lm_huggingface	30.0	32	960.0	221.7	p150
pytorch_Mistral_Small_24B_INSTRUCT_2501_nlp_causal_lm_huggingface	29.0	32	928.0	906.21	p150
pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface	20.0	32	640.0	670.13	n150
pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface	37.0	32	1184.0	319.95	p150
pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface	20.0	32	640.0	657.16	n150
pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface	37.0	32	1184.0	321.29	p150
pytorch_Phi-2_Phi_2_nlp_causal_lm_huggingface	8.0	32	256.0	1494.13	n150
pytorch_Phi-2_Phi_2_nlp_causal_lm_huggingface	20.0	32	640.0	678.99	p150
pytorch_Qwen 2.5 Coder_32B_Instruct_nlp_causal_lm_huggingface	17.0	32	544.0	1452.24	p150
pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface	70.0	32	2240.0	428.19	n150
pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface	126.0	32	4032.0	170.43	p150
pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface	35.0	32	1120.0	479.01	n150
pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface	63.0	32	2016.0	204.92	p150
pytorch_Qwen 2.5_3B_Instruct_nlp_causal_lm_huggingface	30.0	32	960.0	687.51	n150
pytorch_Qwen 2.5_3B_Instruct_nlp_causal_lm_huggingface	56.0	32	1792.0	283.42	p150
pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface	16.0	32	512.0	831.04	n150
pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface	28.0	32	896.0	353.75	p150
pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface	48.0	32	1536.0	1154.14	n150
pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface	96.0	32	3072.0	550.74	p150
pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface	36.0	32	1152.0	744.42	n150
pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface	66.0	32	2112.0	337.78	p150
pytorch_Qwen 3_32B_nlp_causal_lm_huggingface	17.0	32	544.0	1872.69	p150
pytorch_Qwen 3_4B_nlp_causal_lm_huggingface	22.0	32	704.0	1057.1	n150
pytorch_Qwen 3_4B_nlp_causal_lm_huggingface	41.0	32	1312.0	452.22	p150
pytorch_Qwen 3_8B_nlp_causal_lm_huggingface	17.0	32	544.0	1705.14	n150
pytorch_Qwen 3_8B_nlp_causal_lm_huggingface	29.0	32	928.0	739.53	p150

Non-LLM Performance

Model	Batch	Sample/sec	Hardware
Wan2.2-I2V-A14B-DiT	1	0.0	p150
Wan2.2-I2V-A14B-UMT5-Text-Encoder	1	12.0	p150
Wan2.2-I2V-A14B-VAE-Decoder	1	0.0	p150
Wan2.2-I2V-A14B-VAE-Encoder	1	1.0	p150
flux1-dev	1	0.0	p150
flux2	1	0.0	p150
glm-image	1	0.0	p150
hunyuan-image-2.1	1	0.0	p150
janus-pro-1b	1	0.0	n150
janus-pro-1b	1	0.0	p150
janus-pro-7b	1	0.0	p150
playground-v2.5	1	0.0	n150
playground-v2.5	1	0.0	p150
pytorch_BERT_emrecan/bert-base-turkish-cased-mean-nli-stsb-tr_nlp_embed_gen_huggingface	8	159.0	n150
pytorch_BGE-M3_Base_nlp_embed_gen_custom	4	9.0	n150
pytorch_BGE-M3_Base_nlp_embed_gen_custom	4	17.0	p150
pytorch_EfficientNet_Timm_B0_cv_image_cls_timm	8	346.0	n150
pytorch_EfficientNet_Timm_B0_cv_image_cls_timm	8	796.0	p150
pytorch_MNIST_Cnn_Dropout_cv_image_cls_custom	32	13915.0	n150
pytorch_MNIST_Cnn_Dropout_cv_image_cls_custom	32	31693.0	p150
pytorch_MobileNetV2_Mobilenet_v2_cv_image_cls_torch_hub	12	1168.0	n150
pytorch_MobileNetV2_Mobilenet_v2_cv_image_cls_torch_hub	12	2788.0	p150
pytorch_Qwen 3_Embedding_4B_nlp_embed_gen_huggingface	32	46.0	n150
pytorch_Qwen 3_Embedding_4B_nlp_embed_gen_huggingface	32	97.0	p150
pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface	8	1310.0	n150
pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface	8	2767.0	p150
pytorch_SegFormer_B0_Finetuned_Ade_512_512_cv_image_seg_huggingface	1	37.0	n150
pytorch_SegFormer_B0_Finetuned_Ade_512_512_cv_image_seg_huggingface	1	83.0	p150
pytorch_Swin_S_cv_image_cls_torchvision	1	10.0	n150
pytorch_Swin_S_cv_image_cls_torchvision	1	22.0	p150
pytorch_U-Net for Conditional Generation_Base_conditional_generation_huggingface	1	5.0	n150
pytorch_U-Net for Conditional Generation_Base_conditional_generation_huggingface	1	9.0	p150
pytorch_Ultra-Fast Lane Detection v2_TuSimple_ResNet34_Backbone_cv_image_seg_github	1	136.0	n150
pytorch_Ultra-Fast Lane Detection v2_TuSimple_ResNet34_Backbone_cv_image_seg_github	1	236.0	p150
pytorch_VGG19-UNet_base_cv_image_seg_custom	1	150.0	n150
pytorch_VGG19-UNet_base_cv_image_seg_custom	1	310.0	p150
pytorch_ViT_Base_cv_image_cls_huggingface	8	234.0	n150
pytorch_ViT_Base_cv_image_cls_huggingface	8	563.0	p150
pytorch_VoVNet_Ese_Vovnet19b_Dw.ra_In1k_cv_image_cls_timm	8	741.0	n150
pytorch_VoVNet_Ese_Vovnet19b_Dw.ra_In1k_cv_image_cls_timm	8	1548.0	p150
sdxl-lightning	1	0.0	n150
sdxl-lightning	1	0.0	p150
zimage	1	0.0	p150

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
nlp causal lm	Llama	3.1 8B Instruct	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp causal lm	Qwen 3	1 7B	jax	✅	❌	❌	✅	❌	❌	❌	✅	View Source
mm tts	xtts_v2	conditioning	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp embed gen	FLUX	T5TextEncoder	pytorch	✅	❌	❌	❌	❌	✅	❌	❌	Source not available
cv image cls	EfficientNet	B0	pytorch	✅	❌	✅	✅	✅	✅	✅	❌	View Source
cv image seg	Ultra-Fast Lane Detection	TuSimple ResNet18 Backbone	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm tts	xtts_v2	hifigan decoder	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5 Coder	3B Instruct	jax	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	Gemma	2 9B IT	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
cv object det	YOLOv4	Base	pytorch	✅	❌	✅	✅	✅	✅	✅	❌	View Source
mm image ttt	FLUX	Vae	pytorch	✅	❌	❌	❌	❌	✅	❌	❌	Source not available
cv object det	YOLOv7	Default	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5	0.5B Instruct	jax	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	olmo_3	3 1125 32b	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	olmo_3	3 7b think	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
cv image seg	VGG19-UNet	base	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	VoVNet	Ese Vovnet19b Dw.ra In1k	pytorch	✅	❌	✅	✅	✅	✅	✅	❌	View Source
nlp causal lm	Falcon	3 3B Base	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Gemma	1.1 2B IT	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Llama	3.2 1B	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm action prediction	OpenVLA-OFT	Finetuned Libero 10	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
cv image cls	Swin	S	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp embed gen	FLUX	ClipTextEncoder	pytorch	✅	❌	❌	❌	❌	✅	❌	❌	Source not available
cv image cls	DINOv2	Small	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Falcon	3 7B Base	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp causal lm	olmo_3	3 32b think	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
cv image cls	ResNet	ResNet50 HuggingFace High Resolution	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm visual qa	Mistral	base	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp causal lm	Qwen 2.5 Coder	0.5B	jax	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	Phi-3	Mini 128K Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm tts	xtts_v2	speaker encoder	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm image ttt	FLUX	Transformer	pytorch	✅	❌	❌	❌	❌	✅	❌	❌	Source not available
nlp causal lm	GPT-2	Base	jax	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-1.5	Phi 1 5	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5	0.5B Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm image text similarity	SigLIP	Base Patch16 224	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	MNIST	Mlp Custom	jax	❌	✅	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Mistral	Ministral 8B Instruct	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
cv object det	OWL-ViT	Base Patch32	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5	1.5B Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 3	0 6B	pytorch	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	Llama LoRA	3.2 1B	pytorch	❌	✅	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	arcee_ai	arcee Spark	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
nlp causal lm	GPT-OSS	20B	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	olmo_3	3 7b instruct	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-3	Mini 4K Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	MNIST	Mlp Custom 1x2	jax	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	Phi-2	Phi 2	jax	✅	❌	❌	✅	❌	❌	❌	✅	View Source
conditional generation	U-Net for Conditional Generation	Base	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5	3B Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 3	1 7B	pytorch	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	Llama	3.1 70B	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
cv image cls	MNIST	Cnn Dropout	jax	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5	7B Instruct	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp causal lm	Phi-3	Mini Instruct	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 3	4B	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-1 LoRA	Phi 1	pytorch	❌	✅	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Qwen 2.5 Coder	1.5B	jax	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	Qwen 2.5 Coder	1.5B Instruct	jax	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp causal lm	Qwen 3	8B	pytorch	✅	❌	❌	✅	✅	✅	❌	✅	View Source
nlp causal lm	Llama	3.2 3B	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	ViT	Base	pytorch	✅	❌	✅	✅	✅	✅	✅	❌	View Source
nlp embed gen	BERT	emrecan/bert-base-turkish-cased-mean-nli-stsb-tr	pytorch	✅	❌	❌	✅	❌	❌	✅	❌	View Source
cv object det	EfficientDet	D0	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	olmo_3	3 1025 7b	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
nlp embed gen	Qwen 3	Embedding 4B	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv object det	ssd512	ssd512	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv object det	YOLOv9	T	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Mistral	Magistral Small 2506	pytorch	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	Phi-2	Phi 2	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
cv image cls	MNIST	Cnn Batchnorm	jax	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Phi-1	Phi 1	jax	✅	❌	✅	✅	✅	✅	❌	✅	View Source
cv image cls	AlexNet	Custom 1x2	jax	✅	❌	❌	✅	❌	❌	❌	✅	View Source
nlp causal lm	Qwen 3	14B	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
cv object det	YOLO-World	Small 640	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
nlp causal lm	Falcon	3 1B Base	pytorch	✅	❌	✅	✅	✅	✅	❌	✅	View Source
nlp token cls	BiLSTM-CRF	Default	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source
mm visual qa	Llama	3.2 11B Vision	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
nlp causal lm	Mistral	Nemo INSTRUCT 2407	pytorch	✅	❌	❌	❌	✅	✅	❌	❌	View Source
cv object det	DETR	ResNet50 Backbone	pytorch	✅	❌	✅	❌	✅	✅	❌	❌	View Source