`timm` is a deep-learning library created by Ross Wightman and is a collection of SOTA computer vision models, layers, utilities, optimizers, schedulers, data-loaders, augmentations and also training/validating scripts with ability to reproduce ImageNet training results. pip install timm Or for an editable install, git clone https://github.com/rwightman/pytorch-image-models cd pytorch-image-models