PyPI page
Home page
Author:
OT Triton Contributors
Summary:
Sinkhorn optimal transport kernels in PyTorch + Triton (squared Euclidean, no cost matrix materialization).
Latest version:
0.3.3.post1
Required dependencies:
numpy
|
torch
|
triton
Optional dependencies:
geomloss
|
jax
|
matplotlib
|
ott-jax
|
pykeops
|
pytest
Downloads last day:
2
Downloads last week:
39
Downloads last month:
113