issues
search
ott-jax
/
ott
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
https://ott-jax.readthedocs.io
Apache License 2.0
524
stars
80
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
introduce `sinkdiv` simplified API, expand glossary with sinkhorn divergence and LR sinkhorn
#596
marcocuturi
closed
2 days ago
2
Wrong implementation of u_t(x|z) in Brownian Bridge
#595
selmanozleyen
opened
5 days ago
1
add glossary
#594
marcocuturi
closed
1 week ago
3
Can this repository support batched OT calculation?
#593
jsttlgdkycy
closed
1 week ago
4
Onnx support for the ICNNs
#592
malteal
opened
2 weeks ago
0
Was solver and sinkhorn solver handle initializer kwargs inconsistently
#591
selmanozleyen
opened
3 weeks ago
0
Multi GPU Sinkhorn tutorial
#590
michalk8
opened
1 month ago
0
Enable 3.13 CI
#589
michalk8
closed
1 month ago
1
Feature/batched vmap
#588
michalk8
closed
1 month ago
2
fix default for intermediate quadratic layers in ICNN
#587
marcocuturi
closed
1 month ago
1
Switch to `jax.random.key`
#586
michalk8
closed
1 month ago
3
n
#585
marcocuturi
closed
1 month ago
1
expose `SinkhornDivergenceOutput`
#584
marcocuturi
closed
1 month ago
2
Fix dtype promotion in `Sinkhorn`
#583
michalk8
closed
1 month ago
1
Bfloat16/float16 support in Sinkhorn
#582
michalk8
closed
1 month ago
1
Switch to `jax.random.key`
#581
michalk8
closed
1 month ago
0
Feature/refactor was solver
#580
michalk8
closed
1 month ago
2
Refactor/Fix: WassersteinSolver constructor now throws TypeError when an unrecognized argument is given
#579
selmanozleyen
closed
1 month ago
6
automatic scaling of `epsilon` using `std` instead of `mean` by default.
#578
marcocuturi
closed
1 month ago
5
refactor sinkhorn_divergence
#577
marcocuturi
closed
2 months ago
2
implements sliced W
#576
marcocuturi
closed
2 months ago
3
Feature/update prox
#575
michalk8
closed
2 months ago
2
Feature request: Adding `lam` to the regularizer class
#574
pkassraie
closed
2 months ago
0
Vmap-based parallelization possible with variable sized inputs?
#573
sankethvedula
closed
1 month ago
3
update mm-sink tutorial and plot
#572
zoepiran
closed
2 months ago
6
Feature/update progot
#571
michalk8
closed
3 months ago
2
Edits to ProgOT
#570
pkassraie
closed
3 months ago
2
Fix/univariate non ti
#569
michalk8
closed
3 months ago
1
sink div for LR
#568
marcocuturi
closed
3 months ago
5
Feature/progot
#567
michalk8
closed
3 months ago
2
correct convergence criterion for GW
#566
marcocuturi
closed
3 months ago
0
Raise error when instatiating cost/kernel for a grid
#565
michalk8
closed
3 months ago
1
Fix `Grid.prepare_divergences`
#563
michalk8
closed
3 months ago
1
`sinkhorn_divergence` does not handle `Grid` geometry
#562
tlacombe
closed
3 months ago
2
Cannot compute `cost_matrix` or `kernel_matrix` from `geometry.grid.Grid` if both set to `None`
#561
tlacombe
closed
3 months ago
3
Update tutorial structure
#560
michalk8
closed
4 months ago
2
TI transport map
#559
michalk8
closed
4 months ago
1
Fix/gw to lr
#558
michalk8
closed
4 months ago
1
Reproducing results from paper
#557
zwei-beiner
closed
4 months ago
2
Timing comparison with POT
#556
zwei-beiner
closed
4 months ago
6
Fix/quadratic pytree
#555
michalk8
closed
4 months ago
1
Deprecate Python 3.8
#554
michalk8
closed
4 months ago
2
Refactor regularized TI costs
#553
michalk8
closed
4 months ago
2
Pin JAX/jaxlib version
#552
MUCDK
closed
4 months ago
1
Force ICNN to adopt default initialization of its own layers
#551
Algue-Rythme
closed
4 months ago
1
How to calculate 1-Wasserstein distance between two sets of samples?
#550
abhiagwl
closed
4 months ago
2
add MMSinkhorn solver
#549
marcocuturi
closed
4 months ago
3
Fixing typos in documentation
#548
pkassraie
closed
5 months ago
3
Fix low-rank convergence criterion
#547
michalk8
closed
4 months ago
7
Possible bug in `fused_penalty`
#546
gjhuizing
closed
4 months ago
6
Next