torch
einops
transformers
accelerate
fairscale
timm
flash-attn
