tqdm
torch
pillow
numpy
transformers
timm
einops 
flash_attn