Code

Feature/ulrgw#410

Code

Merged

michalk8 merged 49 commits intoott-jax:mainott-jax/ott:mainfrom michalk8:feature/ulrgwmichalk8/ott:feature/ulrgwCopy head branch name to clipboardSep 8, 2023

+1,143-394

Commits

Commits on Aug 2, 2023

• Remove low-rank from GromovWasserstein solver

  michalk8

  committed
  19756c1

  Copy full SHA for 19756c1

• First skeleton loop

  michalk8

  committed
  a079761

  Copy full SHA for a079761

Commits on Aug 3, 2023

• Add LRGW implementation

  michalk8

  committed
  0d0c85f

  Copy full SHA for 0d0c85f

• Add ULFGW

  michalk8

  committed
  1f7f10d

  Copy full SHA for 1f7f10d

• Revert change

  michalk8

  committed
  07aeedf

  Copy full SHA for 07aeedf

• Add a TODO

  michalk8

  committed
  6d257ff

  Copy full SHA for 6d257ff

Commits on Aug 4, 2023

• Fix grad_g in the fused case

  michalk8

  committed
  bf0a50d

  Copy full SHA for bf0a50d

Commits on Aug 17, 2023

• Update docs

  michalk8

  committed
  7b8a0f4

  Copy full SHA for 7b8a0f4

• Remove duplicate citation

  michalk8

  committed
  5855a32

  Copy full SHA for 5855a32

Commits on Aug 18, 2023

• Merge branch 'main' into feature/ulrgw

  michalk8

  committed
  01bd8e6

  Copy full SHA for 01bd8e6

• Fix cost for the fused case

  michalk8

  committed
  e4a70c3

  Copy full SHA for e4a70c3

• Fix bugs in TI

  michalk8

  committed
  4ecdca0

  Copy full SHA for 4ecdca0

• Remove unused import

  michalk8

  committed
  04a3e8b

  Copy full SHA for 04a3e8b

Commits on Aug 31, 2023

• Change way array extraction in LR init works

  michalk8

  committed
  4fb9ee4

  Copy full SHA for 4fb9ee4

• Disallow LR in the old GW solver

  michalk8

  committed
  43d39eb

  Copy full SHA for 43d39eb

• Disallow LR in old GW class

  michalk8

  committed
  0747c56

  Copy full SHA for 0747c56

• Remove is_entropic property

  michalk8

  committed
  83a1c36

  Copy full SHA for 83a1c36

• Use jnp.linalg.norm

  michalk8

  committed
  b783f3a

  Copy full SHA for b783f3a

• Simplify initializers in GW

  michalk8

  committed
  0f661cc

  Copy full SHA for 0f661cc

• Simplify initializer creation for low-rank

  michalk8

  committed
  d79ce31

  Copy full SHA for d79ce31

• Remove temporary name

  michalk8

  committed
  1be699f

  Copy full SHA for 1be699f

• Fix norms

  michalk8

  committed
  87284ae

  Copy full SHA for 87284ae

• Fix linkcheck

  michalk8

  committed
  30f275e

  Copy full SHA for 30f275e

Commits on Sep 1, 2023

• Remove old initializers test

  michalk8

  committed
  f11e19f

  Copy full SHA for f11e19f

• Fix more initializer tests

  michalk8

  committed
  6d4775e

  Copy full SHA for 6d4775e

• Remove LRQuadraticInitializer, reg_ot_cost -> reg_gw_cost

  michalk8

  committed
  fdbafe9

  Copy full SHA for fdbafe9

• host_callback -> io_callback

  michalk8

  committed
  17b447a

  Copy full SHA for 17b447a

• Fix more initializers tests

  michalk8

  committed
  3714057

  Copy full SHA for 3714057

• Fix more tests

  michalk8

  committed
  0fa6be2

  Copy full SHA for 0fa6be2

• Remove initializer mention from the docs

  michalk8

  committed
  3843609

  Copy full SHA for 3843609

• Remove mention of LR initializer

  michalk8

  committed
  62aad99

  Copy full SHA for 62aad99

• Start incorporating GWLoss

  michalk8

  committed
  4e57972

  Copy full SHA for 4e57972

• Simplify reg GW cost computation

  michalk8

  committed
  b312126

  Copy full SHA for b312126

• Finish primal_cost

  michalk8

  committed
  a499102

  Copy full SHA for a499102

• Don't calculate unbal. grads in balanced case

  michalk8

  committed
  4bcb71b

  Copy full SHA for 4bcb71b

• Fix primal_cost in balanced case

  michalk8

  committed
  eea362f

  Copy full SHA for eea362f

• Update GW LR notebook

  michalk8

  committed
  3f14488

  Copy full SHA for 3f14488

Commits on Sep 4, 2023

• Convert quad problem to LR if possible

  michalk8

  committed
  8207696

  Copy full SHA for 8207696

• Convert quad problem to LR if possible

  michalk8

  committed
  ba272f0

  Copy full SHA for ba272f0

Commits on Sep 5, 2023

• Regenerate GWLR Sinkhorn

  michalk8

  committed
  c565ee3

  Copy full SHA for c565ee3

• Regenerate LRSinkhorn

  michalk8

  committed
  f508d5d

  Copy full SHA for f508d5d

• Merge remote-tracking branch 'upstream/main' into feature/ulrgw

  michalk8

  committed
  f7f440f

  Copy full SHA for f7f440f

• Merge branch 'feature/ulrgw' of ssh://github.com/michalk8/ott into feature/ulrgw

  michalk8

  committed
  f703a55

  Copy full SHA for f703a55

• [ci skip] Fix linter

  michalk8

  committed
  ec45825

  Copy full SHA for ec45825

Commits on Sep 7, 2023

• Fix convergence metric

  michalk8

  committed
  68d3ce2

  Copy full SHA for 68d3ce2

• Undo TODO

  michalk8

  committed
  afd8ea6

  Copy full SHA for afd8ea6

• Fix factor

  michalk8

  committed
  e006eae

  Copy full SHA for e006eae

• Regenerate notebooks

  michalk8

  committed
  e06f1b0

  Copy full SHA for e06f1b0

• Add tests

  michalk8

  committed
  f04757f

  Copy full SHA for f04757f