torch>=2.4.1
pytest
numpy
matplotlib
pykeops

[dev]
pytest
black
isort
