numpy
torch>=2.5
triton>=3.1

[dev]
pytest
geomloss
pykeops
jax
ott-jax
matplotlib
