Feature/ Introduce initialization methods for Sinkhorn#98

Merged

michalk8 merged 49 commits intoott-jax:mainfrom JTT94:main

Aug 17, 2022

+682-16

Commits on Jun 30, 2022

add sorting, gaus initializers, add gaus helpers to tools
James Thornton
committed

Commits on Jul 1, 2022

add initialization logic to sinkhorn
James Thornton
committed
remove general ot problem type
James Thornton
committed
remove import tools.gaussian from top level
James Thornton
committed
remove problems from top level
James Thornton
committed
do not register initializer as pytree
James Thornton
committed
add initializer to make
James Thornton
committed
rename init arg to ot_problem
James Thornton
committed
rename init arg to ot_problem
James Thornton
committed
scale gaus init by 2
James Thornton
committed
typo
James Thornton
committed
add basic speed tests
James Thornton
committed
add init to transport tools wrapper, tidy docstring
James Thornton
committed
ceneter potentials in initializers
James Thornton
committed

Commits on Jul 3, 2022

fix lse for null weights
James Thornton
committed
fix flake8 and accidental removal
James Thornton
committed
tidy docstrings
James Thornton
committed
tidy docstrings
James Thornton
committed
docstring flake8
James Thornton
committed

Commits on Jul 4, 2022

flake 8 formatting
James Thornton
committed
Merge branch 'ott-jax:main' into main
JTT94
authored
fix typo
James Thornton
committed
fix stop gradient in Gaussian to include weights and x,y
James Thornton
committed
fix stop gradient in Gaussian to include weights and x,y
James Thornton
committed
fix docstring spaces
James Thornton
committed

Commits on Jul 5, 2022

feedback from initial review
James Thornton
committed
re order local functions before state init
James Thornton
committed

Commits on Jul 6, 2022

optional init_f in sorting init
James Thornton
committed
docstring insert line before return
James Thornton
committed
lint fix
James Thornton
committed

Commits on Jul 12, 2022

incorporate feedback in commit
James Thornton
committed

Commits on Jul 13, 2022

tidy tests, use jax.lax.cond for logic instead of if
James Thornton
committed
add docs, rename sorting initializer
James Thornton
committed
fix merge conflict
James Thornton
committed
resolve test errors in sinkhorn test
James Thornton
committed
fetch upstream for merge
James Thornton
committed

Commits on Jul 14, 2022

incorporate feedback, update tests to pytest, change docstrings, introduce defaultinit class
James Thornton
committed
fix docstring spaces
James Thornton
committed

Commits on Jul 16, 2022

remove spaces and add bibtex
James Thornton
committed

Commits on Aug 17, 2022

add errors for non square cost matrix for sorting, online geoms for initializers, tests
“JTT94”
committed
Merge branch 'main' into main
JTT94
authored
merge fix lint
“JTT94”
committed
merge fix lint
“JTT94”
committed
add initializers as pytees
“JTT94”
committed
add init scaling tests
“JTT94”
committed
add init scaling tests
“JTT94”
committed
simplify vector update flag in sorting initializer
“JTT94”
committed
Fix documentation rendering
michalk8
committed
[ci skip] Fix typo in docs, use fixture in tests
michalk8
committed