more-itertools
numba
numpy
onnxruntime<2,>=1.14
tiktoken
torch
tqdm
triton<3,>=2.0.0

[:platform_machine == "x86_64" and sys_platform == "linux" or sys_platform == "linux2"]
triton<3,>=2.0.0

[dev]
black==24.*
flake8==7.*
nose2
