Repository navigation

optimal transport - #6

Merged

zeakey merged 22 commits into

mainfrom

Jul 13, 2026

Merged

optimal transport#6
zeakey merged 22 commits into
mainfrom
ot

Commits on Apr 2, 2026
- optimal transport
  zeakey
  committed
  eaefbf1
Commits on Apr 22, 2026
- 添加一维成本矩阵构建和Sinkhorn迭代的实现
   maokovski
  committed
  55603aa
Commits on Apr 23, 2026
- debug ot function and add test
  maokovski
  committed
  e668c24
Commits on Apr 24, 2026
- add basic benchmark
  maokovski
  committed
  660ff96
Commits on Apr 28, 2026
- decouple the iteration from the derivation
  maokovski
  committed
  47ce04e
Commits on May 4, 2026
- Implement Sinkhorn iterations in CUDA
  maokovski
  committed
  bc47d27
Commits on May 7, 2026
- Add Sinkhorn CUDA benchmark and correctness tests
  maokovski
  committed
  10d3433
Commits on May 14, 2026
- Add CUDA support for Sinkhorn log-domain iterations and related tests.
  maokovski
  committed
  c631ecf
Commits on May 22, 2026
- Refactor Sinkhorn functions and test gradient accuracy
  maokovski
  committed
  25f6b5d
Commits on Jun 2, 2026
- more test
  maokovski
  committed
  549410a
- try test
  maokovski
  committed
  c0ee5c4
- test on cuda kernal
  maokovski
  committed
  0442882
Commits on Jun 10, 2026
- tighten numeric gradient test and support more dimensions
  maokovski
  committed
  aab8857
- cancel unnecessary change in test.yml
  maokovski
  authored
  a7d80b4
Commits on Jul 8, 2026
- Merge branch 'main' into ot
  zeakey
  committed
  48e1544
- Merge branch 'ot' of github.com:intcomp/torchmorph into ot
  zeakey
  committed
  ba9ad35
- Rework Sinkhorn OT: standard epsilon convention, log-space end-to-end, batched (n, d) API
  
  zeakey
  and
  claude
  committed
  9b55ac4
- Apply black formatting to optimal transport files
  
  zeakey
  and
  claude
  committed
  49d19a0
- Make SinkhornSolver a differentiable nn.Module with automatic device dispatch
  
  zeakey
  and
  claude
  committed
  ec0119e
- Expand Sinkhorn tests to a full device x log_space x dtype matrix
  
  zeakey
  and
  claude
  committed
  2d4da6c
- Optimize Sinkhorn CUDA path: batch tiling, online logsumexp, CUDA graphs
  
  zeakey
  and
  claude
  committed
  ed6d1c3
- Test gradients w.r.t. both source and target
  
  zeakey
  and
  claude
  committed
  3f374ad

Commits

Commits on Apr 2, 2026

Commits on Apr 22, 2026

Commits on Apr 23, 2026

Commits on Apr 24, 2026

Commits on Apr 28, 2026

Commits on May 4, 2026

Commits on May 7, 2026

Commits on May 14, 2026

Commits on May 22, 2026

Commits on Jun 2, 2026

Commits on Jun 10, 2026

Commits on Jul 8, 2026