.codecov.yml
.editorconfig
.flake8
.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
MANIFEST.in
README.md
environment.yml
pyproject.toml
src/ott/__init__.py
src/ott/_version.py
src/ott/types.py
src/ott/utils.py
src/ott/geometry/__init__.py
src/ott/geometry/costs.py
src/ott/geometry/epsilon_scheduler.py
src/ott/geometry/geometry.py
src/ott/geometry/graph.py
src/ott/geometry/grid.py
src/ott/geometry/low_rank.py
src/ott/geometry/pointcloud.py
src/ott/geometry/segment.py
src/ott/initializers/__init__.py
src/ott/initializers/linear/__init__.py
src/ott/initializers/linear/initializers.py
src/ott/initializers/linear/initializers_lr.py
src/ott/initializers/nn/__init__.py
src/ott/initializers/nn/initializers.py
src/ott/initializers/quadratic/__init__.py
src/ott/initializers/quadratic/initializers.py
src/ott/math/__init__.py
src/ott/math/decomposition.py
src/ott/math/fixed_point_loop.py
src/ott/math/matrix_square_root.py
src/ott/math/unbalanced_functions.py
src/ott/math/utils.py
src/ott/problems/__init__.py
src/ott/problems/linear/__init__.py
src/ott/problems/linear/barycenter_problem.py
src/ott/problems/linear/linear_problem.py
src/ott/problems/linear/potentials.py
src/ott/problems/nn/__init__.py
src/ott/problems/nn/dataset.py
src/ott/problems/quadratic/__init__.py
src/ott/problems/quadratic/gw_barycenter.py
src/ott/problems/quadratic/quadratic_costs.py
src/ott/problems/quadratic/quadratic_problem.py
src/ott/solvers/__init__.py
src/ott/solvers/was_solver.py
src/ott/solvers/linear/__init__.py
src/ott/solvers/linear/acceleration.py
src/ott/solvers/linear/continuous_barycenter.py
src/ott/solvers/linear/discrete_barycenter.py
src/ott/solvers/linear/implicit_differentiation.py
src/ott/solvers/linear/sinkhorn.py
src/ott/solvers/linear/sinkhorn_lr.py
src/ott/solvers/nn/__init__.py
src/ott/solvers/nn/conjugate_solvers.py
src/ott/solvers/nn/layers.py
src/ott/solvers/nn/models.py
src/ott/solvers/nn/neuraldual.py
src/ott/solvers/quadratic/__init__.py
src/ott/solvers/quadratic/gromov_wasserstein.py
src/ott/solvers/quadratic/gw_barycenter.py
src/ott/tools/__init__.py
src/ott/tools/k_means.py
src/ott/tools/plot.py
src/ott/tools/segment_sinkhorn.py
src/ott/tools/sinkhorn_divergence.py
src/ott/tools/soft_sort.py
src/ott/tools/gaussian_mixture/__init__.py
src/ott/tools/gaussian_mixture/fit_gmm.py
src/ott/tools/gaussian_mixture/fit_gmm_pair.py
src/ott/tools/gaussian_mixture/gaussian.py
src/ott/tools/gaussian_mixture/gaussian_mixture.py
src/ott/tools/gaussian_mixture/gaussian_mixture_pair.py
src/ott/tools/gaussian_mixture/linalg.py
src/ott/tools/gaussian_mixture/probabilities.py
src/ott/tools/gaussian_mixture/scale_tril.py
src/ott_jax.egg-info/PKG-INFO
src/ott_jax.egg-info/SOURCES.txt
src/ott_jax.egg-info/dependency_links.txt
src/ott_jax.egg-info/requires.txt
src/ott_jax.egg-info/top_level.txt