numpy
torch>=2.5
triton>=3.1

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

[examples]
matplotlib

[notebooks]
matplotlib
jupyterlab
nbclient
