Code

Feature/faster gw init#213

Code

Merged

michalk8 merged 7 commits intoott-jax:mainott-jax/ott:mainfrom michalk8:feature/faster-gw-initmichalk8/ott:feature/faster-gw-initCopy head branch name to clipboardDec 17, 2022

+36-42

Commits

Commits on Dec 16, 2022

• Add faster GW initialization for balanced case

  michalk8

  committed
  537dc57

• Use jax.scipy.special.entr

  michalk8

  committed
  b2d3c70

• Update the unbalanced case

  michalk8

  committed
  47b462a

• Fix wrong xlog

  michalk8

  committed
  c6f40a1

• Parenthesize

  michalk8

  committed
  9d6f903

• Fix imports, change how reg is implemented

  michalk8

  committed
  1bb2eb1

• Transport mass change

  michalk8

  committed
  686943d