torch>=1.10
einops>=0.6.1
pykeops>=2.1.2
