torch==2.4.1
numpy
tiktoken
tqdm
certifi

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