.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
examples/fairness/config.py
examples/fairness/data.py
examples/fairness/losses.py
examples/fairness/main.py
examples/fairness/models.py
examples/fairness/train.py
examples/soft_error/config.py
examples/soft_error/data.py
examples/soft_error/losses.py
examples/soft_error/main.py
examples/soft_error/model.py
examples/soft_error/train.py
ott/__init__.py
ott/_version.py
ott/types.py
ott/utils.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/pointcloud.py
ott/geometry/segment.py
ott/initializers/__init__.py
ott/initializers/linear/__init__.py
ott/initializers/linear/initializers.py
ott/initializers/linear/initializers_lr.py
ott/initializers/nn/__init__.py
ott/initializers/nn/initializers.py
ott/initializers/quadratic/__init__.py
ott/initializers/quadratic/initializers.py
ott/math/__init__.py
ott/math/decomposition.py
ott/math/fixed_point_loop.py
ott/math/matrix_square_root.py
ott/math/unbalanced_functions.py
ott/math/utils.py
ott/problems/__init__.py
ott/problems/linear/__init__.py
ott/problems/linear/barycenter_problem.py
ott/problems/linear/linear_problem.py
ott/problems/linear/potentials.py
ott/problems/quadratic/__init__.py
ott/problems/quadratic/gw_barycenter.py
ott/problems/quadratic/quadratic_costs.py
ott/problems/quadratic/quadratic_problem.py
ott/solvers/__init__.py
ott/solvers/was_solver.py
ott/solvers/linear/__init__.py
ott/solvers/linear/acceleration.py
ott/solvers/linear/continuous_barycenter.py
ott/solvers/linear/discrete_barycenter.py
ott/solvers/linear/implicit_differentiation.py
ott/solvers/linear/sinkhorn.py
ott/solvers/linear/sinkhorn_lr.py
ott/solvers/nn/__init__.py
ott/solvers/nn/icnn.py
ott/solvers/nn/layers.py
ott/solvers/nn/neuraldual.py
ott/solvers/quadratic/__init__.py
ott/solvers/quadratic/gromov_wasserstein.py
ott/solvers/quadratic/gw_barycenter.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