torch==2.4.1
numpy==1.25.2
tiktoken==0.4.0
tqdm
certifi

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