torch>=2.1
numpy>=1.24
safetensors>=0.4
tokenizers>=0.15
tqdm>=4.65

[all]
datasets>=2.14
huggingface_hub>=0.20
gradio>=4.0

[cuda]
torch>=2.1

[dev]
pytest>=7.4
ruff>=0.4

[hf]
datasets>=2.14
huggingface_hub>=0.20

[tpu]
torch_xla>=2.1

[ui]
gradio>=4.0
