torch
torchvision
tqdm
transformers==4.56.1
labml_nn