timm_vit
This directory owns its Python builder, native runtime DSO, bundle section semantics, and tests. It has no source dependency on another family.
Task interfaces
Owned recipes
| Recipe | Checkpoint | Abstract task | Build | Manifest |
|---|---|---|---|---|
timm-vit-base-p16-224-augreg-in21k-ft-in1k | timm/vit_base_patch16_224.augreg_in21k_ft_in1k | classification | fp16; single device | families/timm_vit/tests/manifests/timm-vit-base-p16-224-augreg-in21k-ft-in1k.json |
timm-vit-base-p16-224-augreg-in21k-ft-in1k-tp4 | timm/vit_base_patch16_224.augreg_in21k_ft_in1k | classification | fp32; TP4 | families/timm_vit/tests/manifests/timm-vit-base-p16-224-augreg-in21k-ft-in1k-tp4.json |