torch
einops
torchscale
accelerate
bitsandbytes
fairscale
timm
dataclasses
