torch>=2.1
numpy>=1.24

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

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

[torchneuron]
torch>=2.14
torch-neuronx>=2.14
nki>=0.6
