numpy>=1.22.0
safetensors>=0.4.0
huggingface_hub>=0.20.0

[cuda]
cupy-cuda12x>=13.0.0
triton>=3.0.0

[cuda:platform_system == "Linux"]
nvidia-nccl-cu12>=2.18.1
nvidia-cublas-cu12>=12.1.0
nvidia-cuda-runtime-cu12>=12.1.0
nvidia-cuda-nvrtc-cu12>=12.1.0

[test]
pytest>=7.0.0
