torch==2.8.0
pydantic>=2.0.0
safetensors>=0.5.0
flashinfer-python>=0.3.1

[cuda12]
nvidia-cublas-cu12>=12.3
nvidia-cudnn-cu12>=9.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=22.0.0
isort>=5.0.0
mypy>=1.0.0
ruff>=0.1.0
pre-commit>=3.0.0
pydoc-markdown>=4.8.2
