.codecov.yml
.editorconfig
.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
CITATION.bib
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
src/ott/__init__.py
src/ott/_version.py
src/ott/datasets.py
src/ott/py.typed
src/ott/types.py
src/ott/utils.py
src/ott/experimental/__init__.py
src/ott/experimental/mmsinkhorn.py
src/ott/geometry/__init__.py
src/ott/geometry/costs.py
src/ott/geometry/distrib_costs.py
src/ott/geometry/epsilon_scheduler.py
src/ott/geometry/geodesic.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/regularizers.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/neural/__init__.py
src/ott/initializers/neural/meta_initializer.py
src/ott/initializers/quadratic/__init__.py
src/ott/initializers/quadratic/initializers.py
src/ott/math/__init__.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/neural/__init__.py
src/ott/neural/datasets.py
src/ott/neural/methods/__init__.py
src/ott/neural/methods/expectile_neural_dual.py
src/ott/neural/methods/monge_gap.py
src/ott/neural/methods/neuraldual.py
src/ott/neural/methods/flows/__init__.py
src/ott/neural/methods/flows/dynamics.py
src/ott/neural/methods/flows/genot.py
src/ott/neural/methods/flows/otfm.py
src/ott/neural/networks/__init__.py
src/ott/neural/networks/icnn.py
src/ott/neural/networks/potentials.py
src/ott/neural/networks/velocity_field.py
src/ott/neural/networks/layers/__init__.py
src/ott/neural/networks/layers/conjugate.py
src/ott/neural/networks/layers/posdef.py
src/ott/neural/networks/layers/time_encoder.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/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/utils.py
src/ott/solvers/was_solver.py
src/ott/solvers/linear/__init__.py
src/ott/solvers/linear/_solve.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/lineax_implicit.py
src/ott/solvers/linear/lr_utils.py
src/ott/solvers/linear/sinkhorn.py
src/ott/solvers/linear/sinkhorn_lr.py
src/ott/solvers/linear/univariate.py
src/ott/solvers/quadratic/__init__.py
src/ott/solvers/quadratic/_solve.py
src/ott/solvers/quadratic/gromov_wasserstein.py
src/ott/solvers/quadratic/gromov_wasserstein_lr.py
src/ott/solvers/quadratic/gw_barycenter.py
src/ott/solvers/quadratic/lower_bound.py
src/ott/tools/__init__.py
src/ott/tools/k_means.py
src/ott/tools/plot.py
src/ott/tools/progot.py
src/ott/tools/segment_sinkhorn.py
src/ott/tools/sinkhorn_divergence.py
src/ott/tools/sliced.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