LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
ott/__init__.py
ott/version.py
ott/core/__init__.py
ott/core/anderson.py
ott/core/bar_problems.py
ott/core/continuous_barycenter.py
ott/core/dataclasses.py
ott/core/discrete_barycenter.py
ott/core/fixed_point_loop.py
ott/core/gromov_wasserstein.py
ott/core/icnn.py
ott/core/implicit_differentiation.py
ott/core/momentum.py
ott/core/neuraldual.py
ott/core/problems.py
ott/core/quad_problems.py
ott/core/segment.py
ott/core/sinkhorn.py
ott/core/sinkhorn_lr.py
ott/core/unbalanced_functions.py
ott/core/was_solver.py
ott/geometry/__init__.py
ott/geometry/costs.py
ott/geometry/epsilon_scheduler.py
ott/geometry/geometry.py
ott/geometry/grid.py
ott/geometry/low_rank.py
ott/geometry/matrix_square_root.py
ott/geometry/ops.py
ott/geometry/pointcloud.py
ott/tools/__init__.py
ott/tools/plot.py
ott/tools/sinkhorn_divergence.py
ott/tools/soft_sort.py
ott/tools/transport.py
ott/tools/gaussian_mixture/__init__.py
ott/tools/gaussian_mixture/fit_gmm.py
ott/tools/gaussian_mixture/fit_gmm_pair.py
ott/tools/gaussian_mixture/gaussian.py
ott/tools/gaussian_mixture/gaussian_mixture.py
ott/tools/gaussian_mixture/gaussian_mixture_pair.py
ott/tools/gaussian_mixture/linalg.py
ott/tools/gaussian_mixture/probabilities.py
ott/tools/gaussian_mixture/scale_tril.py
ott_jax.egg-info/PKG-INFO
ott_jax.egg-info/SOURCES.txt
ott_jax.egg-info/dependency_links.txt
ott_jax.egg-info/requires.txt
ott_jax.egg-info/top_level.txt