torch
transformers[timm]
matplotlib