1.5.0.dev20260818002307
·
2 commits
to main
since this release
Installation
Via PyPI
pip install tt-forge==1.5.0.dev20260818002307 --extra-index-url https://pypi.eng.aws.tenstorrent.com/Via Docker
docker pull ghcr.io/tenstorrent/tt-forge-slim:1.5.0.dev20260818002307Dependency commits
tt-xla commit: 9004e0475e8a7a539927ebfa751f7a16b5114862
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.dev20260818002307
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 | 878.04 | p150 |
| pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface | 54.0 | 32 | 1728.0 | 700.17 | n150 |
| pytorch_Falcon_3_1B_Base_nlp_causal_lm_huggingface | 104.0 | 32 | 3328.0 | 297.96 | p150 |
| pytorch_Falcon_3_3B_Base_nlp_causal_lm_huggingface | 35.0 | 32 | 1120.0 | 880.85 | n150 |
| pytorch_Falcon_3_3B_Base_nlp_causal_lm_huggingface | 66.0 | 32 | 2112.0 | 395.66 | p150 |
| pytorch_Falcon_3_7B_Base_nlp_causal_lm_huggingface | 18.0 | 32 | 576.0 | 1184.94 | n150 |
| pytorch_Falcon_3_7B_Base_nlp_causal_lm_huggingface | 35.0 | 32 | 1120.0 | 499.98 | p150 |
| pytorch_GPT-OSS_120B_nlp_causal_lm_huggingface | 4.0 | 8 | 32.0 | 1406.48 | p150 |
| pytorch_GPT-OSS_20B_nlp_causal_lm_huggingface | 8.0 | 64 | 512.0 | 7444.98 | n150 |
| pytorch_GPT-OSS_20B_nlp_causal_lm_huggingface | 21.0 | 1 | 21.0 | 310.54 | p150 |
| pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface | 38.0 | 32 | 1216.0 | 636.21 | n150 |
| pytorch_Gemma_1.1_2B_IT_nlp_causal_lm_huggingface | 74.0 | 32 | 2368.0 | 241.25 | p150 |
| pytorch_Llama_3.1_70B_Instruct_nlp_causal_lm_huggingface | 4.0 | 32 | 128.0 | 11855.54 | n150 |
| pytorch_Llama_3.1_70B_Instruct_nlp_causal_lm_huggingface | 12.0 | 32 | 384.0 | 1814.75 | p150 |
| pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface | 21.0 | 32 | 672.0 | 1305.29 | n150 |
| pytorch_Llama_3.1_8B_Instruct_nlp_causal_lm_huggingface | 38.0 | 32 | 1216.0 | 579.8 | p150 |
| pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface | 64.0 | 32 | 2048.0 | 556.74 | n150 |
| pytorch_Llama_3.2_1B_Instruct_nlp_causal_lm_huggingface | 125.0 | 32 | 4000.0 | 234.98 | p150 |
| pytorch_Llama_3.2_3B_Instruct_nlp_causal_lm_huggingface | 30.0 | 32 | 960.0 | 612.66 | n150 |
| pytorch_Llama_3.2_3B_Instruct_nlp_causal_lm_huggingface | 55.0 | 32 | 1760.0 | 284.71 | p150 |
| pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface | 20.0 | 32 | 640.0 | 1264.46 | n150 |
| pytorch_Mistral_7B_INSTRUCT_v03_nlp_causal_lm_huggingface | 35.0 | 32 | 1120.0 | 579.97 | p150 |
| pytorch_Mistral_Ministral_8B_Instruct_nlp_causal_lm_huggingface | 14.0 | 32 | 448.0 | 577.54 | n150 |
| pytorch_Mistral_Ministral_8B_Instruct_nlp_causal_lm_huggingface | 30.0 | 32 | 960.0 | 223.27 | p150 |
| pytorch_Mistral_Small_24B_INSTRUCT_2501_nlp_causal_lm_huggingface | 29.0 | 32 | 928.0 | 900.96 | p150 |
| pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface | 20.0 | 32 | 640.0 | 652.92 | n150 |
| pytorch_Phi-1.5_Phi_1_5_nlp_causal_lm_huggingface | 37.0 | 32 | 1184.0 | 326.23 | p150 |
| pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface | 20.0 | 32 | 640.0 | 667.78 | n150 |
| pytorch_Phi-1_Phi_1_nlp_causal_lm_huggingface | 37.0 | 32 | 1184.0 | 317.93 | p150 |
| pytorch_Phi-2_Phi_2_nlp_causal_lm_huggingface | 8.0 | 32 | 256.0 | 1432.09 | n150 |
| pytorch_Phi-2_Phi_2_nlp_causal_lm_huggingface | 20.0 | 32 | 640.0 | 684.97 | p150 |
| pytorch_Qwen 2.5 Coder_32B_Instruct_nlp_causal_lm_huggingface | 17.0 | 32 | 544.0 | 1480.68 | p150 |
| pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface | 69.0 | 32 | 2208.0 | 409.87 | n150 |
| pytorch_Qwen 2.5_0.5B_Instruct_nlp_causal_lm_huggingface | 126.0 | 32 | 4032.0 | 170.06 | p150 |
| pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface | 35.0 | 32 | 1120.0 | 484.16 | n150 |
| pytorch_Qwen 2.5_1.5B_Instruct_nlp_causal_lm_huggingface | 63.0 | 32 | 2016.0 | 202.07 | p150 |
| pytorch_Qwen 2.5_3B_Instruct_nlp_causal_lm_huggingface | 30.0 | 32 | 960.0 | 709.46 | n150 |
| pytorch_Qwen 2.5_3B_Instruct_nlp_causal_lm_huggingface | 57.0 | 32 | 1824.0 | 283.75 | p150 |
| pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface | 16.0 | 32 | 512.0 | 808.65 | n150 |
| pytorch_Qwen 2.5_7B_Instruct_nlp_causal_lm_huggingface | 27.0 | 32 | 864.0 | 351.71 | p150 |
| pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface | 50.0 | 32 | 1600.0 | 1171.65 | n150 |
| pytorch_Qwen 3_0_6B_nlp_causal_lm_huggingface | 95.0 | 32 | 3040.0 | 549.04 | p150 |
| pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface | 36.0 | 32 | 1152.0 | 742.55 | n150 |
| pytorch_Qwen 3_1_7B_nlp_causal_lm_huggingface | 66.0 | 32 | 2112.0 | 335.67 | p150 |
| pytorch_Qwen 3_32B_nlp_causal_lm_huggingface | 17.0 | 32 | 544.0 | 1842.82 | p150 |
| pytorch_Qwen 3_4B_nlp_causal_lm_huggingface | 23.0 | 32 | 736.0 | 995.94 | n150 |
| pytorch_Qwen 3_8B_nlp_causal_lm_huggingface | 16.0 | 32 | 512.0 | 1655.33 | n150 |
| pytorch_Qwen 3_8B_nlp_causal_lm_huggingface | 30.0 | 32 | 960.0 | 731.07 | p150 |
Non-LLM Performance
| Model | Batch | Sample/sec | Hardware |
|---|---|---|---|
| HunyuanVideo-1.5-480p-t2v-distilled | 1 | 0.0 | p150 |
| 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 | 161.0 | n150 |
| pytorch_BGE-M3_Base_nlp_embed_gen_custom | 4 | 8.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 | 14025.0 | n150 |
| pytorch_MNIST_Cnn_Dropout_cv_image_cls_custom | 32 | 30944.0 | p150 |
| pytorch_MobileNetV2_Mobilenet_v2_cv_image_cls_torch_hub | 12 | 1169.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 | 97.0 | p150 |
| pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface | 8 | 1313.0 | n150 |
| pytorch_ResNet_ResNet50_HuggingFace_cv_image_cls_huggingface | 8 | 2762.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 | 135.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 | 149.0 | n150 |
| pytorch_VGG19-UNet_base_cv_image_seg_custom | 1 | 309.0 | p150 |
| pytorch_ViT_Base_cv_image_cls_huggingface | 8 | 234.0 | n150 |
| pytorch_ViT_Base_cv_image_cls_huggingface | 8 | 562.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 | 1551.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 |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| cv image cls | MNIST | Mlp Custom | jax | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv object det | OWL-ViT | Base Patch32 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv object det | YOLOv7 | Default | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Phi-2 | Phi 2 | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| mm action prediction | OpenVLA-OFT | Finetuned Libero 10 | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| mm tts | xtts_v2 | gpt decode | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Phi-1.5 | Phi 1 5 | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 3 | 1 7B | pytorch | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv image cls | ViT | Base | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 1.5B Instruct | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | ALLaM | 7B Instruct | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 3 | 14B | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 3 | 4B | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | olmo_3 | 3 7b think | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 32B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| mm image ttt | FLUX | Vae | pytorch | ✅ | ❌ | ❌ | ❌ | ❌ | ✅ | ❌ | ❌ | Source not available |
| nlp causal lm | Mistral | Ministral 8B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| cv panoptic seg | Panoptic Segmentation | ResNet50 Backbone 1x COCO | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Gemma LoRA | 1.1 2B IT | pytorch | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 3B Instruct | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Phi-1 | Phi 1 | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | GPT-OSS | 20B | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 Coder | 1.5B Instruct | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv image cls | DINOv2 | Small | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Llama | 3.1 8B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv image cls | MNIST | Cnn Dropout | jax | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 Coder | 0.5B | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv object det | YOLOS Small | Small | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 3B Instruct | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| cv image cls | VoVNet | Ese Vovnet19b Dw.ra In1k | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv object det | YOLOP | Default | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | AlexNet | Custom 1x2 | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Falcon | 3 1B Base | pytorch | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 Coder | 3B Instruct | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| conditional generation | U-Net for Conditional Generation | Base | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 0.5B Instruct | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 Coder | 1.5B | jax | ✅ | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | olmo_3 | 3 7b instruct | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 14B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp embed gen | Qwen 3 | Embedding 4B | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp token cls | BiLSTM-CRF | Default | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | ResNet | ResNet50 HuggingFace High Resolution | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp embed gen | FLUX | ClipTextEncoder | pytorch | ✅ | ❌ | ❌ | ❌ | ❌ | ✅ | ❌ | ❌ | Source not available |
| nlp causal lm | Gemma | 2 27B IT | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Mistral | Devstral Small 2505 | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 2.5 Coder | 3B | jax | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Qwen 3 | 8B | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Llama | 3.2 1B | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 7B Instruct | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv image seg | Ultra-Fast Lane Detection | TuSimple ResNet18 Backbone | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | VGG | HF Vgg19 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | MobileNetV1 | Mobilenet v1 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| mm tts | xtts_v2 | hifigan decoder | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Mistral | Nemo INSTRUCT 2407 | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| mm tts | xtts_v2 | conditioning | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| mm visual qa | Llama | 3.2 11B Vision | pytorch | ✅ | ❌ | ❌ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image seg | MaskFormer Swin-B | Swin Base Coco | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Phi-1 | Phi 1 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 3 | 4B | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | GPT-2 | Base | jax | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | MNIST | Cnn Batchnorm | jax | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image seg | VGG19-UNet | base | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Qwen 2.5 | 0.5B Instruct | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | olmo_3 | 3 32b think | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| cv object det | YOLO-World | Small 640 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Gemma | 1.1 2B IT | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Falcon | 3 7B Base | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Llama | 3.2 3B | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| mm tts | xtts_v2 | gpt prefill | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Llama LoRA | 3.2 1B | pytorch | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Phi-3 | Mini 128K Instruct | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Falcon | 3 3B Base | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | SOLAR-10_7B | 10 7B Instruct v1.0 | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| cv image cls | SegFormer | Mit B0 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv object det | ssd512 | ssd512 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| cv image cls | MNIST | Mlp Custom | jax | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Gemma | 2 9B IT | pytorch | ✅ | ❌ | ❌ | ✅ | ✅ | ✅ | ❌ | ✅ | View Source |
| nlp causal lm | Mistral | Magistral Small 2506 | pytorch | ✅ | ❌ | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ | View Source |
| nlp causal lm | Phi-2 | Phi 2 | pytorch | ✅ | ❌ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | View Source |
| nlp causal lm | Llama | 3.1 70B Instruct | pytorch | ✅ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ✅ | View Source |