torch>=2.1
numpy>=1.24

[dev]
pytest>=7.0
pytest-benchmark>=4.0
scipy>=1.11
mkdocs>=1.5
mkdocs-material>=9.0

[neuron]
neuronxcc>=2.24
torch-neuronx>=2.9
