torch>=2.1
numpy>=1.24

[dev]
pytest>=7.0
pytest-benchmark>=4.0
pytest-cov>=4.0
ruff>=0.6
pre-commit>=3.5
scipy>=1.11
mkdocs>=1.5
mkdocs-material>=9.0

[neuron]
nki>=0.3.0
torch-neuronx>=2.9
