torch
pytest
numpy
matplotlib
pykeops

[dev]
pytest
black
isort
