matplotlib
scikit-image
timm
einops
transformers