torch
torchaudio
transformers
pytorch_lightning