Code

Potentials as tuple#488

Code

Merged

michalk8 merged 9 commits intomainott-jax/ott:mainfrom feature/potentials-as-tupleott-jax/ott:feature/potentials-as-tupleCopy head branch name to clipboardFeb 8, 2024

+448-464

Commits

Commits on Feb 8, 2024

• Add support for .potentials for SinkhornOutput

  michalk8

  committed
  720f2f4

  Copy full SHA for 720f2f4

• Remove explicit dtype casting

  michalk8

  committed
  ac3cdc6

  Copy full SHA for ac3cdc6

• Add leading or trailing 0s

  michalk8

  committed
  4fdbfaf

  Copy full SHA for 4fdbfaf

• Fix dtype in test

  michalk8

  committed
  993da56

  Copy full SHA for 993da56

• Fix typo

  michalk8

  committed
  17f3be7

  Copy full SHA for 17f3be7

• Remove jnp.atleast_1d in subsetting

  michalk8

  committed
  b0ed993

  Copy full SHA for b0ed993

• Fix scalar subsetting

  michalk8

  committed
  c8edbd3

  Copy full SHA for c8edbd3

• Fix flax=0.8.1 docs typos in docs linter

  michalk8

  committed
  2c9a292

  Copy full SHA for 2c9a292

• Fix more pet-peeves

  michalk8

  committed
  70c5495

  Copy full SHA for 70c5495