Timm compatibility

[c-mulliken]

We need to build out full compatibility with timm models. I anticipate some activation shape issues for certain model classes.

[c-mulliken]

definition of done

utils/load_model.py supports loading timm models by name via config (e.g. timm: true + model_name: "resnet50" or similar).

activation extraction works for a small, explicit suite of timm models covering the annoying families:

resnet (easy)

vit (token/cls shapes)

convnext (stages)

swin (hierarchical tokens)

tests: add a cpu-only test that:

instantiates each model

runs a dummy batch through your activation pipeline

asserts expected shapes / no crashes

docs: update README/config examples to show timm usage.

ci passes.

guardrails

do not change benchmark metrics logic unless needed

prefer minimal invasive changes

if a model can’t be supported, document it + skip with a clear reason