torch>=2.1
numpy>=1.24

[dev]
pytest>=7.0
pytest-benchmark>=4.0
pytest-cov>=4.1
scipy>=1.11

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