torch
torchvision
torchaudio
datasets
transformers
accelerate
fast-hadamard-transform