1.5.0.dev20260805004653
·
1 commit
to main
since this release
Installation
Via PyPI
pip install tt-forge==1.5.0.dev20260805004653 --extra-index-url https://pypi.eng.aws.tenstorrent.com/Via Docker
docker pull ghcr.io/tenstorrent/tt-forge-slim:1.5.0.dev20260805004653Dependency commits
tt-xla commit: a585859cb550ff77d8fbeafb3443464388287973
tt-mlir commit: 724d2ff7b48da2fa73977e85476f6da1c835434b
tt-metal commit: f1f4ff75579ebd7a69c7da52d45368f273026d85
What's Changed
- Uplift third_party/tt_forge_models to 8734a136b439fc3261145a411acbc038ca108e04 2026-08-01 by @vvukomanTT in #1023
Full Changelog: 1.5.0.dev20260801001900...1.5.0.dev20260805004653
LLM Performance
| Model | Token/sec/user | Batch | Token/sec | ttft (ms) | Hardware |
|---|---|---|---|---|---|
| pytorch_DeepSeek-V3.2_deepseek_v3_2_exp_modified_nlp_causal_lm_custom | 3.0 | 128 | 384.0 | 8200.77 | n150 |
| pytorch_Falcon_3_10B_Base_nlp_causal_lm_huggingface | 40.0 | 32 | 1280.0 | 847.84 | p150 |
| pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface | 55.0 | 32 | 1760.0 | 647.7 | n150 |
| pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface | 101.0 | 32 | 3232.0 | 295.38 | p150 |
| pytorch_Falcon_3_3B_Base_nlp_causal_lm_huggingface | 36.0 | 32 | 1152.0 | 808.81 | n150 |
| pytorch_Falcon_3_3B_Base_nlp_causal_lm_huggingface | 67.0 | 32 | 2144.0 | 381.07 | p150 |
| pytorch_Falcon_3_7B_Base_nlp_causal_lm_huggingface | 18.0 | 32 | 576.0 | 1211.2 | n150 |
| pytorch_Falcon_3_7B_Base_nlp_causal_lm_huggingface | 35.0 | 32 | 1120.0 | 496.26 | p150 |
| pytorch_GLM_4.7_nlp_causal_lm_huggingface | 7.0 | 128 | 896.0 | 2692.01 | n150 |
| pytorch_GPT-OSS_20B_nlp_causal_lm_huggingface | 8.0 | 64 | 512.0 | 6927.3 | n150 |
| pytorch_GPT-OSS_20B_nlp_causal_lm_huggingface | 21.0 | 1 | 21.0 | 340.48 | p150 |
| pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface | 38.0 | 32 | 1216.0 | 636.06 | n150 |
| pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface | 78.0 | 32 | 2496.0 | 240.73 | p150 |
| pytorch_Kimi-K2.6_kimi_k2_6_modified_nlp_causal_lm_custom | 6.0 | 64 | 384.0 | 3580.82 | n150 |
| pytorch_Kimi-K2_kimi_k2_instruct_modified_nlp_causal_lm_custom | 6.0 | 64 | 384.0 | 3406.06 | n150 |
| pytorch_Llama_3.1_70B_Instruct_nlp_causal_lm_huggingface | 12.0 | 32 | 384.0 | 1804.48 | p150 |
| pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface | 22.0 | 32 | 704.0 | 1229.52 | n150 |
| pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface | 50.0 | 32 | 1600.0 | 650.61 | p150 |
| pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface | 64.0 | 32 | 2048.0 | 540.73 | n150 |
| pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface | 122.0 | 32 | 3904.0 | 232.37 | p150 |
| pytorch_Llama_3.2_3B_Instruct_nlp_causal_lm_huggingface | 30.0 | 32 | 960.0 | 583.55 | n150 |
| pytorch_Llama_3.2_3B_Instruct_nlp_causal_lm_huggingface | 53.0 | 32 | 1696.0 | 282.79 | p150 |
| pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface | 19.0 | 32 | 608.0 | 1286.36 | n150 |
| pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface | 35.0 | 32 | 1120.0 | 551.3 | p150 |
| pytorch_Mistral_Small_24B_INSTRUCT_2501_nlp_causal_lm_huggingface | 29.0 | 32 | 928.0 | 894.11 | p150 |
| pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface | 20.0 | 32 | 640.0 | 661.98 | n150 |
| pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface | 37.0 | 32 | 1184.0 | 327.42 | p150 |
| pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface | 20.0 | 32 | 640.0 | 634.25 | n150 |
| pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface | 37.0 | 32 | 1184.0 | 322.61 | p150 |
| pytorch_Phi-2_Phi_2_nlp_causal_lm_huggingface | 8.0 | 32 | 256.0 | 1402.74 | n150 |
| pytorch_Phi-2_Phi_2_nlp_causal_lm_huggingface | 20.0 | 32 | 640.0 | 698.64 | p150 |
| pytorch_Qwen 2.5 Coder_32B_Instruct_nlp_causal_lm_huggingface | 17.0 | 32 | 544.0 | 1433.05 | p150 |
| pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface | 69.0 | 32 | 2208.0 | 400.53 | n150 |
| pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface | 126.0 | 32 | 4032.0 | 179.71 | p150 |
| pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface | 36.0 | 32 | 1152.0 | 484.45 | n150 |
| pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface | 62.0 | 32 | 1984.0 | 209.79 | p150 |
| pytorch_Qwen 2.5_3B_Instruct_nlp_causal_lm_huggingface | 30.0 | 32 | 960.0 | 662.83 | n150 |
| pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface | 16.0 | 32 | 512.0 | 810.04 | n150 |
| pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface | 27.0 | 32 | 864.0 | 355.38 | p150 |
| pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface | 49.0 | 32 | 1568.0 | 1094.78 | n150 |
| pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface | 98.0 | 32 | 3136.0 | 518.65 | p150 |
| pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface | 36.0 | 32 | 1152.0 | 680.81 | n150 |
| pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface | 67.0 | 32 | 2144.0 | 319.69 | p150 |
| pytorch_Qwen 3_32B_nlp_causal_lm_huggingface | 17.0 | 32 | 544.0 | 1779.92 | p150 |
| pytorch_Qwen 3_4B_nlp_causal_lm_huggingface | 22.0 | 32 | 704.0 | 933.28 | n150 |
| pytorch_Qwen 3_4B_nlp_causal_lm_huggingface | 42.0 | 32 | 1344.0 | 456.21 | p150 |
| pytorch_Qwen 3_8B_nlp_causal_lm_huggingface | 17.0 | 32 | 544.0 | 1516.18 | n150 |
| pytorch_Qwen 3_8B_nlp_causal_lm_huggingface | 29.0 | 32 | 928.0 | 713.8 | 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 | 2.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 | 161.0 | n150 |
| pytorch_BGE-M3_Base_nlp_embed_gen_custom | 4 | 9.0 | n150 |
| pytorch_BGE-M3_Base_nlp_embed_gen_custom | 4 | 18.0 | p150 |
| pytorch_EfficientNet_Timm_B0_cv_image_cls_timm | 8 | 352.0 | n150 |
| pytorch_EfficientNet_Timm_B0_cv_image_cls_timm | 8 | 797.0 | p150 |
| pytorch_MNIST_Cnn_Dropout_cv_image_cls_custom | 32 | 14046.0 | n150 |
| pytorch_MNIST_Cnn_Dropout_cv_image_cls_custom | 32 | 27642.0 | p150 |
| pytorch_MobileNetV2_Mobilenet_v2_cv_image_cls_torch_hub | 12 | 1241.0 | n150 |
| pytorch_MobileNetV2_Mobilenet_v2_cv_image_cls_torch_hub | 12 | 2912.0 | p150 |
| pytorch_Qwen 3_Embedding_4B_nlp_embed_gen_huggingface | 32 | 48.0 | n150 |
| pytorch_Qwen 3_Embedding_4B_nlp_embed_gen_huggingface | 32 | 109.0 | p150 |
| pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface | 8 | 1351.0 | n150 |
| pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface | 8 | 2750.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 | 80.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 | 135.0 | n150 |
| pytorch_Ultra-Fast Lane Detection v2_TuSimple_ResNet34_Backbone_cv_image_seg_github | 1 | 237.0 | p150 |
| pytorch_VGG19-UNet_base_cv_image_seg_custom | 1 | 147.0 | n150 |
| pytorch_VGG19-UNet_base_cv_image_seg_custom | 1 | 305.0 | p150 |
| pytorch_ViT_Base_cv_image_cls_huggingface | 8 | 235.0 | n150 |
| pytorch_ViT_Base_cv_image_cls_huggingface | 8 | 566.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 | 1569.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.5 Coder | 0.5B | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv img to img | Autoencoder | linear | pytorch | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp embed gen | BERT | emrecan/bert-base-turkish-cased-mean-nli-stsb-tr | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ✅ | ❌ | View Source |
| nlp causal lm | Qwen 3 | 4B | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | AlexNet | Custom 1x2 | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 Coder | 1.5B Instruct | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | 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 |
| cv object det | OWL-ViT | Base Patch32 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 32B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Falcon | 3 10B Base | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp embed gen | FLUX | ClipTextEncoder | pytorch | ✅ | ❌ | ❌ | ❌ | ❌ | ✅ | ❌ | ❌ | Source not available |
| nlp token cls | BiLSTM-CRF | Default | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Mistral | 7B INSTRUCT v03 | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv object det | YOLOv4 | Base | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | View Source |
| nlp causal lm | Phi-2 | Phi 2 | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 | 3B | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | arcee_ai | arcee Spark | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 7B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | SOLAR-10_7B | 10 7B Instruct v1.0 | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| conditional generation | U-Net for Conditional Generation | Base | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv object det | YOLOS Small | Small | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp embed gen | FLUX | T5TextEncoder | pytorch | ✅ | ❌ | ❌ | ❌ | ❌ | ✅ | ❌ | ❌ | Source not available |
| nlp causal lm | Phi-2 | Phi 2 | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 3 | 14B | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv object det | ssd512 | ssd512 | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | MNIST | Mlp Custom | jax | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | MNIST | Mlp Custom 1x2 | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| cv image cls | MobileNetV2 | Mobilenet v2 | pytorch | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ | View Source |
| cv image cls | ResNet | ResNet50 HuggingFace High Resolution | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 1.5B Instruct | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Llama | 3.2 3B | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | VGG | HF Vgg19 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | MNIST | Cnn Nodropout | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv object det | YOLOv9 | T | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| mm image text similarity | CLIP | Base Patch16 | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Llama | 3.1 8B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Gemma | 2 27B IT | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Llama LoRA | 3.2 1B | pytorch | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | GPT-2 | Xl | jax | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 0.5B Instruct | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 3 | 0 6B | pytorch | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv object det | DETR | ResNet50 Backbone | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | MobileNetV1 | Mobilenet v1 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Phi-3 | Mini 4K Instruct | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Phi-4 | Phi 4 | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 3 | 8B | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 | 3B Instruct | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 Coder | 3B | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| cv object det | Attention DenseUNet | Base | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | olmo_3 | 3 1125 32b | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| cv object det | YOLOv7 | Default | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Gemma | 1.1 7B IT | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | GPT-OSS | 20B | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Phi-1 LoRA | Phi 1 | pytorch | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Falcon | 3 1B Base | pytorch | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Falcon | 3 7B Base | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 Coder | 32B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Mistral | Large INSTRUCT 2411 | pytorch | ✅ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Gemma | 1.1 2B IT | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Phi-1 | Phi 1 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv object det | PointPillars | pointpillars | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 72B | pytorch | ✅ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 | 72B Instruct | pytorch | ✅ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ✅ | View Source |
| cv image cls | MNIST | Cnn Dropout | jax | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Phi-1 | Phi 1 | 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 |
| cv panoptic seg | Panoptic Segmentation | ResNet50 Backbone 1x COCO | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 14B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv image cls | MNIST | Cnn Batchnorm | jax | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 Coder | 1.5B | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 | 1.5B Instruct | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | GPT-2 | Base | jax | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Mistral | Nemo INSTRUCT 2407 | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| cv image seg | VGG19-UNet | base | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Mistral | Devstral Small 2505 | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| cv image cls | SegFormer | Mit B0 | pytorch | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ | View Source |
| cv image cls | MNIST | Mlp Custom | jax | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv object det | TransFuser | None | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |