LICENSE
MANIFEST.in
README.md
pyproject.toml
tests/test_analysis.py
tests/test_color_transfer.py
tests/test_data.py
tests/test_experiments.py
tests/test_gaussian_mixture_barycenter_generator.py
tests/test_gaussian_mixture_generator.py
tests/test_mnist.py
tests/test_problems.py
tests/test_problems_exponential_generator.py
tests/test_solvers.py
tests/test_utils_generate_nd_grid.py
tests/test_utils_generator_helpers_exponential.py
tests/test_utils_generator_helpers_gaussian.py
tests/test_utils_generator_to_weights_list.py
uot/__init__.py
uot/algorithms/col_gen.py
uot/algorithms/gradient_ascent.py
uot/algorithms/lbfgs.py
uot/algorithms/lp.py
uot/algorithms/pdgd.py
uot/algorithms/sinkhorn.py
uot/algorithms/utils.py
uot/algorithms/rapdhg/__init__.py
uot/algorithms/rapdhg/iteration_stats_utils.py
uot/algorithms/rapdhg/loop_utils.py
uot/algorithms/rapdhg/mp_io.py
uot/algorithms/rapdhg/preprocess.py
uot/algorithms/rapdhg/rapdhg.py
uot/algorithms/rapdhg/restart.py
uot/algorithms/rapdhg/solver_log.py
uot/algorithms/rapdhg/termination.py
uot/algorithms/rapdhg/utils.py
uot/algorithms/rapdhg/strategies/reg_strategy.py
uot/algorithms/rapdhg/strategies/update_strategy.py
uot/analysis/__init__.py
uot/analysis/analysis.py
uot/analysis/plotter.py
uot/analysis/run_tests.py
uot/analysis/split_results.py
uot/data/__init__.py
uot/data/dataset_loader.py
uot/data/measure.py
uot/experiments/__init__.py
uot/experiments/experiment.py
uot/experiments/measurement.py
uot/experiments/runner.py
uot/experiments/real_data/color_transfer/color_transfer.py
uot/experiments/real_data/color_transfer/color_transfer_metrics.py
uot/experiments/real_data/color_transfer/color_transfer_problem.py
uot/experiments/real_data/color_transfer/experiment.py
uot/experiments/real_data/color_transfer/image_data.py
uot/experiments/real_data/color_transfer/measurement.py
uot/experiments/real_data/color_transfer/measurement_helpers.py
uot/experiments/real_data/color_transfer/runner.py
uot/experiments/real_data/color_transfer/utils.py
uot/experiments/real_data/color_transfer/visualization.py
uot/experiments/real_data/mnist_classification/count_pairwise_distances.py
uot/experiments/real_data/mnist_classification/mnist_classification.py
uot/experiments/synthetic/benchmark.py
uot/experiments/synthetic/benchmark_gaussians.py
uot/problems/__init__.py
uot/problems/barycenter_problem.py
uot/problems/base_problem.py
uot/problems/hdf5_store.py
uot/problems/inspect_store.py
uot/problems/iterator.py
uot/problems/multi_marginal.py
uot/problems/problem_generator.py
uot/problems/problem_serializer.py
uot/problems/store.py
uot/problems/two_marginal.py
uot/problems/generators/__init__.py
uot/problems/generators/cauchy_generator.py
uot/problems/generators/exponential_generator.py
uot/problems/generators/gaussian_mixture_barycenter_generator.py
uot/problems/generators/gaussian_mixture_generator.py
uot/problems/generators/generalized_hyperbolic_mixture.py
uot/problems/generators/independent_cauchy.py
uot/problems/generators/independent_exponential.py
uot/problems/generators/paired_generator.py
uot/problems/generators/students_t.py
uot/problems/generators/toy_barycenter_generator.py
uot/solvers/__init__.py
uot/solvers/base_solver.py
uot/solvers/linear_programming.py
uot/solvers/pdlp.py
uot/solvers/pdlp_barycenter.py
uot/solvers/solver.py
uot/solvers/solver_config.py
uot/solvers/back_and_forth/__init__.py
uot/solvers/back_and_forth/barycenter.py
uot/solvers/back_and_forth/c_transform.py
uot/solvers/back_and_forth/forward_pushforward.py
uot/solvers/back_and_forth/map_pushforward.py
uot/solvers/back_and_forth/method.py
uot/solvers/back_and_forth/monge_map.py
uot/solvers/back_and_forth/pushforward.py
uot/solvers/back_and_forth/solver.py
uot/solvers/gradient_ascent/__init__.py
uot/solvers/gradient_ascent/_make_schedule.py
uot/solvers/gradient_ascent/_smith_best_lr.py
uot/solvers/gradient_ascent/adam.py
uot/solvers/gradient_ascent/amsgrad.py
uot/solvers/gradient_ascent/gradient_ascent.py
uot/solvers/gradient_ascent/gradient_plain.py
uot/solvers/gradient_ascent/saga.py
uot/solvers/lbfgs/__init__.py
uot/solvers/lbfgs/lbfgs.py
uot/solvers/lbfgs/lbfgs_pure.py
uot/solvers/lbfgs/lbfgs_warmstart.py
uot/solvers/pdlp_bary/__init__.py
uot/solvers/pdlp_bary/iteration_stats_utils.py
uot/solvers/pdlp_bary/loop_utils.py
uot/solvers/pdlp_bary/mp_io.py
uot/solvers/pdlp_bary/preprocess.py
uot/solvers/pdlp_bary/rapdhg.py
uot/solvers/pdlp_bary/restart.py
uot/solvers/pdlp_bary/solver_log.py
uot/solvers/pdlp_bary/termination.py
uot/solvers/pdlp_bary/utils.py
uot/solvers/pdlp_bary/strategies/reg_strategy.py
uot/solvers/pdlp_bary/strategies/update_strategy.py
uot/solvers/sinkhorn/__init__.py
uot/solvers/sinkhorn/barycenter.py
uot/solvers/sinkhorn/divergence.py
uot/solvers/sinkhorn/sinkhorn.py
uot/solvers/sinkhorn/sinkhorn_log.py
uot/utils/__init__.py
uot/utils/build_measure.py
uot/utils/central_gradient_nd.py
uot/utils/compute_sparsity.py
uot/utils/costs.py
uot/utils/exceptions.py
uot/utils/generate_nd_grid.py
uot/utils/generator_to_weights_list.py
uot/utils/import_helpers.py
uot/utils/instantiate_solver.py
uot/utils/logging.py
uot/utils/maps.py
uot/utils/mnist_helpers.py
uot/utils/notebook_helpers.py
uot/utils/plotting.py
uot/utils/problems_loaders.py
uot/utils/solver_helpers.py
uot/utils/types.py
uot/utils/yaml_helpers.py
uot/utils/generator_helpers/__init__.py
uot/utils/generator_helpers/cauchy.py
uot/utils/generator_helpers/exponential.py
uot/utils/generator_helpers/gaussian.py
uot/utils/generator_helpers/get_axes.py
uot/utils/generator_helpers/shapes.py
uot/utils/metrics/__init__.py
uot/utils/metrics/convexity.py
uot/utils/metrics/finite_diff.py
uot/utils/metrics/interpolation.py
uot/utils/metrics/pushforward_map_metrics.py
uot_bench.egg-info/PKG-INFO
uot_bench.egg-info/SOURCES.txt
uot_bench.egg-info/dependency_links.txt
uot_bench.egg-info/requires.txt
uot_bench.egg-info/top_level.txt