LICENSE
README.md
pyproject.toml
tests/test_hub_package.py
torch-ext/flash_sinkhorn/__init__.py
torch-ext/flash_sinkhorn/_autograd.py
torch-ext/flash_sinkhorn/c_transform.py
torch-ext/flash_sinkhorn/cg.py
torch-ext/flash_sinkhorn/hvp.py
torch-ext/flash_sinkhorn/implicit_grad.py
torch-ext/flash_sinkhorn/samples_loss.py
torch-ext/flash_sinkhorn/sinkhorn_solvers.py
torch-ext/flash_sinkhorn.egg-info/PKG-INFO
torch-ext/flash_sinkhorn.egg-info/SOURCES.txt
torch-ext/flash_sinkhorn.egg-info/dependency_links.txt
torch-ext/flash_sinkhorn.egg-info/requires.txt
torch-ext/flash_sinkhorn.egg-info/top_level.txt
torch-ext/flash_sinkhorn/bench/__init__.py
torch-ext/flash_sinkhorn/bench/bench_backward.py
torch-ext/flash_sinkhorn/bench/bench_forward.py
torch-ext/flash_sinkhorn/kernels/__init__.py
torch-ext/flash_sinkhorn/kernels/_common.py
torch-ext/flash_sinkhorn/kernels/_triton_helpers.py
torch-ext/flash_sinkhorn/kernels/apply_flash.py
torch-ext/flash_sinkhorn/kernels/apply_ott.py
torch-ext/flash_sinkhorn/kernels/c_transform_sqeuclid.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_blocksparse_sqeuclid.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_flashstyle_sqeuclid.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_splitn_sqeuclid.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_triton_apply_fused_sqeuclid.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_triton_apply_sqeuclid.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_triton_cg_dense.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_triton_cg_python_batched.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_triton_geomloss_sqeuclid.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_triton_grad_sqeuclid.py
torch-ext/flash_sinkhorn/kernels/sinkhorn_triton_ott_sqeuclid.py
torch-ext/flash_sinkhorn/multiscale/__init__.py
torch-ext/flash_sinkhorn/multiscale/_preprocess.py
torch-ext/flash_sinkhorn/multiscale/block_mask.py
torch-ext/flash_sinkhorn/multiscale/check.py
torch-ext/flash_sinkhorn/multiscale/coarse.py
torch-ext/flash_sinkhorn/multiscale/envelope.py
torch-ext/flash_sinkhorn/multiscale/fine.py
torch-ext/flash_sinkhorn/multiscale/geometry.py
torch-ext/flash_sinkhorn/multiscale/mask_refine.py
torch-ext/flash_sinkhorn/multiscale/solver.py
torch-ext/flash_sinkhorn/multiscale/sparse_step.py
torch-ext/flash_sinkhorn/testing/__init__.py
torch-ext/flash_sinkhorn/testing/reference_hvp.py
torch-ext/flash_sinkhorn/testing/reference_sinkhorn.py
torch-ext/flash_sinkhorn/testing/test_apply_plan_flashstyle.py
torch-ext/flash_sinkhorn/testing/test_c_transform.py
torch-ext/flash_sinkhorn/testing/test_flashstyle_parity.py
torch-ext/flash_sinkhorn/testing/test_geomloss_sinkhorn_triton.py
torch-ext/flash_sinkhorn/testing/test_geomloss_vs_triton.py
torch-ext/flash_sinkhorn/testing/test_half_cost.py
torch-ext/flash_sinkhorn/testing/test_hvp_parity.py
torch-ext/flash_sinkhorn/testing/test_hvp_sqeuclid.py
torch-ext/flash_sinkhorn/testing/test_multiscale_kernels.py
torch-ext/flash_sinkhorn/testing/test_multiscale_solver.py
torch-ext/flash_sinkhorn/testing/test_ott_vs_triton.py
torch-ext/flash_sinkhorn/testing/test_pad_to_multiple.py
torch-ext/flash_sinkhorn/testing/test_samples_loss_api.py
torch-ext/flash_sinkhorn/testing/test_semi_unbalanced_forward.py
torch-ext/flash_sinkhorn/testing/test_sinkhorn_triton.py
torch-ext/flash_sinkhorn/testing/test_unbalanced_sinkhorn.py