torch
einops
flash-attn
accelerate
bitsandbytes
fairscale
timm
flamingo-pytorch
