LICENSE
README.md
pyproject.toml
setup.py
sources/torchmatch/__init__.py
sources/torchmatch/__init__.pyi
sources/torchmatch/py.typed
sources/torchmatch.egg-info/PKG-INFO
sources/torchmatch.egg-info/SOURCES.txt
sources/torchmatch.egg-info/dependency_links.txt
sources/torchmatch.egg-info/requires.txt
sources/torchmatch.egg-info/top_level.txt
sources/torchmatch/assignment/__init__.py
sources/torchmatch/assignment/_auction.py
sources/torchmatch/assignment/_cost.py
sources/torchmatch/assignment/_cpu.py
sources/torchmatch/assignment/_cuda.py
sources/torchmatch/assignment/_greedy.py
sources/torchmatch/assignment/_loader.py
sources/torchmatch/assignment/_solve.py
sources/torchmatch/assignment/ops.py
sources/torchmatch/assignment/ops.pyi
sources/torchmatch/assignment/py.typed
sources/torchmatch/assignment/cpu/jonker_compact_avx2.h
sources/torchmatch/assignment/cpu/jonker_compact_core.h
sources/torchmatch/assignment/cpu/jonker_dense_avx2.h
sources/torchmatch/assignment/cpu/jonker_dense_core.cpp
sources/torchmatch/assignment/cpu/jonker_dense_core.h
sources/torchmatch/assignment/cpu/jonker_dense_flat.h
sources/torchmatch/assignment/cpu/jonker_scalar.cpp
sources/torchmatch/assignment/cpu/jonker_scalar.h
sources/torchmatch/assignment/cpu/ops.cpp
sources/torchmatch/assignment/cuda/control.cuh
sources/torchmatch/assignment/cuda/cuda_common.cuh
sources/torchmatch/assignment/cuda/dispatch.cu
sources/torchmatch/assignment/cuda/hybrid.cu
sources/torchmatch/assignment/cuda/jonker_tiled.cuh
sources/torchmatch/assignment/cuda/lawler.cu
sources/torchmatch/assignment/cuda/munkres.cu
sources/torchmatch/assignment/cuda/ops.cpp
sources/torchmatch/assignment/cuda/workspace.cuh
sources/torchmatch/bench/__init__.py
sources/torchmatch/bench/__main__.py
sources/torchmatch/bench/_collect.py
sources/torchmatch/bench/_filename.py
sources/torchmatch/bench/_machine.py
sources/torchmatch/bench/_scrub.py
sources/torchmatch/transport/__init__.py
sources/torchmatch/transport/_loader.py
sources/torchmatch/transport/py.typed
sources/torchmatch/transport/matrix/__init__.py
sources/torchmatch/transport/matrix/_cpu.py
sources/torchmatch/transport/matrix/_cuda.py
sources/torchmatch/transport/matrix/_exact_emd.py
sources/torchmatch/transport/matrix/_log_sinkhorn.py
sources/torchmatch/transport/matrix/_schedule.py
sources/torchmatch/transport/matrix/_sinkhorn_divergence.py
sources/torchmatch/transport/matrix/_solve.py
sources/torchmatch/transport/matrix/_unbalanced_sinkhorn.py
sources/torchmatch/transport/matrix/_validate.py
sources/torchmatch/transport/matrix/ops.py
sources/torchmatch/transport/matrix/ops.pyi
sources/torchmatch/transport/matrix/cpu/exact_emd_op.cpp
sources/torchmatch/transport/matrix/cpu/ops.cpp
sources/torchmatch/transport/matrix/cpu/exact/EMD.h
sources/torchmatch/transport/matrix/cpu/exact/EMD_wrapper.cpp
sources/torchmatch/transport/matrix/cpu/exact/core.h
sources/torchmatch/transport/matrix/cpu/exact/full_bipartitegraph.h
sources/torchmatch/transport/matrix/cpu/exact/network_simplex_simple.h
sources/torchmatch/transport/matrix/cpu/exact/sparse_bipartitegraph.h
sources/torchmatch/transport/matrix/cuda/ops.cpp
sources/torchmatch/transport/samples/__init__.py
sources/torchmatch/transport/samples/_autograd.py
sources/torchmatch/transport/samples/_c_transform.py
sources/torchmatch/transport/samples/_cg.py
sources/torchmatch/transport/samples/_hvp.py
sources/torchmatch/transport/samples/_implicit_grad.py
sources/torchmatch/transport/samples/_loss.py
sources/torchmatch/transport/samples/_solvers.py
sources/torchmatch/transport/samples/kernels/__init__.py
sources/torchmatch/transport/samples/kernels/_common.py
sources/torchmatch/transport/samples/kernels/_triton_helpers.py
sources/torchmatch/transport/samples/kernels/apply_fused_sqeuclid.py
sources/torchmatch/transport/samples/kernels/apply_raw.py
sources/torchmatch/transport/samples/kernels/apply_shifted.py
sources/torchmatch/transport/samples/kernels/apply_sqeuclid.py
sources/torchmatch/transport/samples/kernels/c_transform_sqeuclid.py
sources/torchmatch/transport/samples/kernels/cg_dense.py
sources/torchmatch/transport/samples/kernels/cg_python_batched.py
sources/torchmatch/transport/samples/kernels/grad_sqeuclid.py
sources/torchmatch/transport/samples/kernels/streaming_sqeuclid.py
tests/test_assignment_greedy.py
tests/test_lap_inf_mask.py
tests/test_lap_jonker_batch.py
tests/test_lap_jonker_float32.py
tests/test_lap_jonker_scalar.py
tests/test_lap_jonker_single.py
tests/test_lap_opcheck.py
tests/test_match_dispatch.py
tests/test_transport_exact_emd.py
tests/test_transport_integration.py
tests/test_transport_log_sinkhorn.py
tests/test_transport_log_sinkhorn_grad.py
tests/test_transport_matrix_namespace.py
tests/test_transport_matrix_solve.py
tests/test_transport_opcheck.py
tests/test_transport_samples_autograd.py
tests/test_transport_samples_imports.py
tests/test_transport_samples_loss.py
tests/test_transport_schedule.py
tests/test_transport_sinkhorn_divergence.py
tests/test_transport_smoke.py
tests/test_transport_unbalanced_sinkhorn.py
tests/test_transport_validate.py