python >= 3.8
pytorch >= 1.12.1  # For scatter_reduce
torchvision        # With matching version for your pytorch install
timm == 0.4.12     # Might work on other versions, but this is what we tested
torchprofile       # For FLOPs calculation
tqdm               # To display a progress bar 