torch>=2.1
numpy>=1.24

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

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