numpy
tqdm
gpuctypes

[:platform_system == "Darwin"]
pyobjc-framework-Metal
pyobjc-framework-libdispatch

[arm]
unicorn

[linting]
pylint
mypy
typing-extensions
pre-commit
ruff
types-tqdm

[llvm]
llvmlite

[testing]
torch
pillow
pytest
pytest-xdist
onnx==1.15.0
onnx2torch
opencv-python
tabulate
safetensors
transformers
sentencepiece
tiktoken
librosa
networkx
hypothesis

[triton]
triton-nightly>=2.1.0.dev20231014192330
