LICENSE
README.md
pyproject.toml
src/flash_sinkhorn/__init__.py
src/flash_sinkhorn/_autograd.py
src/flash_sinkhorn/cg.py
src/flash_sinkhorn/hvp.py
src/flash_sinkhorn/implicit_grad.py
src/flash_sinkhorn/samples_loss.py
src/flash_sinkhorn/sinkhorn_solvers.py
src/flash_sinkhorn.egg-info/PKG-INFO
src/flash_sinkhorn.egg-info/SOURCES.txt
src/flash_sinkhorn.egg-info/dependency_links.txt
src/flash_sinkhorn.egg-info/requires.txt
src/flash_sinkhorn.egg-info/top_level.txt
src/flash_sinkhorn/bench/__init__.py
src/flash_sinkhorn/bench/bench_backward.py
src/flash_sinkhorn/bench/bench_forward.py
src/flash_sinkhorn/kernels/__init__.py
src/flash_sinkhorn/kernels/_common.py
src/flash_sinkhorn/kernels/_triton_helpers.py
src/flash_sinkhorn/kernels/apply_flash.py
src/flash_sinkhorn/kernels/apply_ott.py
src/flash_sinkhorn/kernels/sinkhorn_flashstyle_sqeuclid.py
src/flash_sinkhorn/kernels/sinkhorn_triton_apply_fused_sqeuclid.py
src/flash_sinkhorn/kernels/sinkhorn_triton_apply_sqeuclid.py
src/flash_sinkhorn/kernels/sinkhorn_triton_cg_dense.py
src/flash_sinkhorn/kernels/sinkhorn_triton_cg_python_batched.py
src/flash_sinkhorn/kernels/sinkhorn_triton_geomloss_sqeuclid.py
src/flash_sinkhorn/kernels/sinkhorn_triton_grad_sqeuclid.py
src/flash_sinkhorn/kernels/sinkhorn_triton_ott_sqeuclid.py
src/flash_sinkhorn/testing/__init__.py
src/flash_sinkhorn/testing/reference_hvp.py
src/flash_sinkhorn/testing/reference_sinkhorn.py
src/flash_sinkhorn/testing/test_apply_plan_flashstyle.py
src/flash_sinkhorn/testing/test_flashstyle_parity.py
src/flash_sinkhorn/testing/test_geomloss_sinkhorn_triton.py
src/flash_sinkhorn/testing/test_geomloss_vs_triton.py
src/flash_sinkhorn/testing/test_half_cost.py
src/flash_sinkhorn/testing/test_hvp_parity.py
src/flash_sinkhorn/testing/test_hvp_sqeuclid.py
src/flash_sinkhorn/testing/test_ott_vs_triton.py
src/flash_sinkhorn/testing/test_samples_loss_api.py
src/flash_sinkhorn/testing/test_semi_unbalanced_forward.py
src/flash_sinkhorn/testing/test_sinkhorn_triton.py
src/flash_sinkhorn/testing/test_unbalanced_sinkhorn.py