numpy>=1.19.5
pytest
scikit_learn>=1.6.0
scipy>=1.14.0
torch
array_api_compat
