LICENSE
README.md
pyproject.toml
tests/test_distances.py
tests/test_embedding.py
tests/test_graph.py
tests/test_lowrank.py
tests/test_multiscale.py
tests/test_sampling.py
tests/test_sinkhorn.py
tests/test_sinkhorn_grad.py
tests/test_solver.py
tests/test_spiral_swissroll.py
tests/test_triton_sinkhorn.py
torchgw/__init__.py
torchgw/_distances.py
torchgw/_embedding.py
torchgw/_graph.py
torchgw/_lowrank.py
torchgw/_multiscale.py
torchgw/_sampling.py
torchgw/_solver.py
torchgw/_triton_sinkhorn.py
torchgw/_utils.py
torchgw.egg-info/PKG-INFO
torchgw.egg-info/SOURCES.txt
torchgw.egg-info/dependency_links.txt
torchgw.egg-info/requires.txt
torchgw.egg-info/top_level.txt