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:
4
Downloads last week:
45
Downloads last month:
157