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