Code

Remove assertions in cost functions#340

Code

Merged

michalk8 merged 7 commits intoott-jax:mainott-jax/ott:mainfrom michalk8:fix/costs-assertionsmichalk8/ott:fix/costs-assertionsCopy head branch name to clipboardMar 22, 2023

+23-29

Commits

Commits on Mar 22, 2023

• Remove assertions

  michalk8

  committed
  be8b38b

• Use jnp.where

  michalk8

  committed
  b51d805

• Fix missing 2nd jnp.where

  michalk8

  committed
  0761e89

• Fix PointCloud link in tutorial

  michalk8

  committed
  0b3c477

• Use jax.lax.cond

  michalk8

  committed
  9c8fec6

• Don't make p in costs traceable

  michalk8

  committed
  1454b39

• Fix children not being iterable

  michalk8

  committed
  c73beb7