torch
einops
torchscale
transformers
accelerate
bitsandbytes
fairscale
timm
dataclasses
