torch>=2.1
numpy
jinja2
filelock
wrapt
packaging
coloredlogs

[all]
tamm[sentencepiece,torchvision,transformers,xxhash]

[sentencepiece]
sentencepiece

[torchvision]
torchvision

[transformers]
transformers
safetensors

[xxhash]
xxhash
