.codecov.yml
.editorconfig
.flake8
.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
.style.yapf
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.cfg
ott/__init__.py
ott/_version.py
ott/core/__init__.py
ott/core/_math_utils.py
ott/core/anderson.py
ott/core/bar_problems.py
ott/core/continuous_barycenter.py
ott/core/dataclasses.py
ott/core/decomposition.py
ott/core/discrete_barycenter.py
ott/core/fixed_point_loop.py
ott/core/gromov_wasserstein.py
ott/core/gw_barycenter.py
ott/core/icnn.py
ott/core/implicit_differentiation.py
ott/core/initializers.py
ott/core/initializers_lr.py
ott/core/layers.py
ott/core/linear_problems.py
ott/core/momentum.py
ott/core/neuraldual.py
ott/core/potentials.py
ott/core/problems.py
ott/core/quad_initializers.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/examples/fairness/config.py
ott/examples/fairness/data.py
ott/examples/fairness/losses.py
ott/examples/fairness/main.py
ott/examples/fairness/models.py
ott/examples/fairness/train.py
ott/examples/soft_error/config.py
ott/examples/soft_error/data.py
ott/examples/soft_error/losses.py
ott/examples/soft_error/main.py
ott/examples/soft_error/model.py
ott/examples/soft_error/train.py
ott/geometry/__init__.py
ott/geometry/costs.py
ott/geometry/epsilon_scheduler.py
ott/geometry/geometry.py
ott/geometry/graph.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/k_means.py
ott/tools/plot.py
ott/tools/segment_sinkhorn.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/not-zip-safe
ott_jax.egg-info/requires.txt
ott_jax.egg-info/top_level.txt