PyPI page
Home page
Author:
OT Triton Contributors
License:
MIT
Summary:
Sinkhorn optimal transport kernels in PyTorch + Triton (squared Euclidean, no cost matrix materialization).
Latest version:
0.4.1
Required dependencies:
numpy
|
torch
|
triton
Optional dependencies:
geomloss
|
jax
|
jupyterlab
|
matplotlib
|
nbclient
|
ott-jax
|
pykeops
|
pytest
Downloads last day:
6
Downloads last week:
25
Downloads last month:
251
7-day smoothing