torch
numpy
tqdm

[:platform_system == "Linux" and platform_machine == "x86_64"]
torch-xla==2.7.0
