numpy
torch>=2.5
triton>=3.1

[dev]
pytest
geomloss>=0.3
pykeops
jax
ott-jax
matplotlib
