Index A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | Q | R | S | T | U | V | W | X A a (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput attribute) A_comp (ott.geometry.regularizers.Quadratic property) a_s (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) all_pairs() (ott.geometry.costs.Arccos method) (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.EuclideanP method) (ott.geometry.costs.NegDotProduct method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.RegTICost method) (ott.geometry.costs.SoftDTW method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) (ott.geometry.costs.UnbalancedBures method) (ott.geometry.distrib_costs.UnivariateWasserstein method) alphas (ott.tools.progot.ProgOTOutput attribute) AndersonAcceleration (class in ott.solvers.linear.acceleration) animate() (ott.tools.plot.Plot method) (ott.tools.plot.PlotMM method) apply() (ott.neural.networks.potentials.BasePotential method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) apply_cost() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_fn (ott.neural.networks.potentials.PotentialTrainState attribute) apply_gradients() (ott.neural.networks.potentials.PotentialTrainState method) apply_kernel() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_lse_kernel() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_square_cost() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_transport_from_potentials() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) apply_transport_from_scalings() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) Arccos (class in ott.geometry.costs) assignment (ott.tools.k_means.KMeansOutput attribute) B b (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput attribute) barycenter() (ott.geometry.costs.Arccos method) (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.EuclideanP method) (ott.geometry.costs.NegDotProduct method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.RegTICost method) (ott.geometry.costs.SoftDTW method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) (ott.geometry.costs.UnbalancedBures method) (ott.geometry.distrib_costs.UnivariateWasserstein method) (ott.geometry.pointcloud.PointCloud method) BasePotential (class in ott.neural.networks.potentials) batch_size (ott.geometry.pointcloud.PointCloud property) (ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader attribute) (ott.solvers.linear.semidiscrete.SemidiscreteSolver attribute) batched_vmap() (in module ott.utils) Benamou-Brenier bias (ott.geometry.low_rank.LRCGeometry property) Brenier potential Brenier theorem Bures (class in ott.geometry.costs) C c-concave function c-transform calibrate() (ott.tools.conformal.OTCP method) calibration_scores (ott.tools.conformal.OTCP attribute) callback (ott.solvers.linear.semidiscrete.SemidiscreteSolver attribute) can_LRC (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) centroids (ott.tools.k_means.KMeansOutput attribute) cholesky (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) clone() (ott.neural.networks.potentials.BasePotential method) component_weight_ob (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) component_weights (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) components() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) compute_error() (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) compute_kl_reg_cost() (ott.solvers.linear.sinkhorn.SinkhornState method) compute_reg_gw_cost() (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) compute_reg_ot_cost() (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) conditional_log_prob() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) ConjugateResults (class in ott.neural.networks.layers.conjugate) constant_epsilon_scheduler() (in module ott.solvers.linear.semidiscrete) converged (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) (ott.solvers.linear.semidiscrete.SemidiscreteOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) (ott.tools.k_means.KMeansOutput attribute) (ott.tools.progot.ProgOTOutput property) (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput attribute) copy() (ott.neural.networks.potentials.BasePotential method) copy_epsilon() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) Cosine (class in ott.geometry.costs) cost (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) cost_1 (ott.geometry.low_rank.LRCGeometry property) cost_2 (ott.geometry.low_rank.LRCGeometry property) cost_at_geom() (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) cost_fn (ott.neural.data.ot_dataloader.LinearOTDataloader attribute) (ott.neural.methods.expectile_neural_dual.ENOTPotentials attribute) (ott.problems.linear.potentials.DualPotentials attribute) cost_fns (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) cost_matrix (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) cost_rank (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) cost_t (ott.experimental.mmsinkhorn.MMSinkhornOutput property) cost_unbalanced_correction() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) CostFn (class in ott.geometry.costs) costs (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) costs_bary (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) count() (ott.experimental.mmsinkhorn.MMSinkhornOutput method) (ott.neural.networks.layers.conjugate.ConjugateResults method) (ott.problems.quadratic.quadratic_costs.GWLoss method) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState method) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn.SinkhornState method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) (ott.solvers.linear.univariate.UnivariateOutput method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState method) (ott.tools.k_means.KMeansOutput method) (ott.tools.progot.ProgOTOutput method) coupling covariance (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) covariance() (ott.tools.gaussian_mixture.gaussian.Gaussian method) covariance_fixpoint_iter() (ott.geometry.costs.Bures method) create() (ott.neural.networks.potentials.PotentialTrainState class method) create_train_state() (ott.neural.networks.potentials.BasePotential method) crossed_threshold (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) curvature() (in module ott.neural.methods.flow_matching) D dataset (ott.neural.data.ot_dataloader.LinearOTDataloader attribute) DEFAULT_EPSILON_SCALE (in module ott.geometry.epsilon_scheduler) default_progress_fn() (in module ott.utils) DefaultInitializer (class in ott.initializers.linear.initializers) diag (ott.solvers.linear.sinkhorn.SinkhornOutput property) diag_cost (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) distance() (ott.neural.methods.expectile_neural_dual.ENOTPotentials method) (ott.problems.linear.potentials.DualPotentials method) divergence (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput attribute) dtype (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) (ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud property) (ott.problems.linear.linear_problem.LinearProblem property) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) dual Kantorovich potentials dual Kantorovich problem dual transportation polyhedron dual_a (ott.solvers.linear.univariate.UnivariateOutput attribute) dual_b (ott.solvers.linear.univariate.UnivariateOutput attribute) dual_cost (ott.solvers.linear.semidiscrete.HardAssignmentOutput property) (ott.solvers.linear.sinkhorn.SinkhornOutput property) dual_costs (ott.solvers.linear.univariate.UnivariateOutput property) dualizing DualPotentials (class in ott.problems.linear.potentials) dykstra_update_kernel() (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) dykstra_update_lse() (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) E EMA (class in ott.neural.networks.velocity_field.ema) ENOTPotentials (class in ott.neural.methods.expectile_neural_dual) ent_reg_cost (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput property) entropic map entropy (ott.solvers.linear.sinkhorn.SinkhornOutput property) entropy-regularized optimal transport envelope theorem Epsilon (class in ott.geometry.epsilon_scheduler) epsilon (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) (ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud property) (ott.neural.data.ot_dataloader.LinearOTDataloader attribute) (ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader attribute) (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem property) (ott.solvers.linear.semidiscrete.SemidiscreteState attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) epsilon_scheduler (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) epsilon_scheduler() (ott.solvers.linear.semidiscrete.SemidiscreteSolver method) epsilons (ott.tools.progot.ProgOTOutput attribute) error (ott.tools.k_means.KMeansOutput attribute) error_batch_size (ott.solvers.linear.semidiscrete.SemidiscreteSolver attribute) error_eval_every (ott.solvers.linear.semidiscrete.SemidiscreteSolver attribute) error_num_repeats (ott.solvers.linear.semidiscrete.SemidiscreteSolver attribute) error_threshold (ott.solvers.linear.acceleration.Momentum attribute) errors (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) (ott.solvers.linear.semidiscrete.SemidiscreteOutput attribute) (ott.solvers.linear.semidiscrete.SemidiscreteState attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornState attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput attribute) Euclidean (class in ott.geometry.costs) EuclideanP (class in ott.geometry.costs) eval() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.icnn.KeyNet method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.MLP method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.ema.EMA method) (ott.neural.networks.velocity_field.mlp.MLP method) (ott.neural.networks.velocity_field.unet.UNet method) evaluate_velocity_field() (in module ott.neural.methods.flow_matching) ExpectileNeuralDual (class in ott.neural.methods.expectile_neural_dual) extrapolation() (ott.solvers.linear.acceleration.AndersonAcceleration method) F f (ott.neural.methods.expectile_neural_dual.ENOTPotentials attribute) (ott.problems.linear.potentials.DualPotentials attribute) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) (ott.solvers.linear.semidiscrete.HardAssignmentOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput property) f1 (ott.problems.quadratic.quadratic_costs.GWLoss attribute) f2 (ott.problems.quadratic.quadratic_costs.GWLoss attribute) f_potential() (ott.tools.gaussian_mixture.gaussian.Gaussian method) factors (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput attribute) FenchelConjugateLBFGS (class in ott.neural.networks.layers.conjugate) FenchelConjugateSolver (class in ott.neural.networks.layers.conjugate) first_order_conditions() (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) fit_model_em() (in module ott.tools.gaussian_mixture.fit_gmm) fit_transport() (ott.tools.conformal.OTCP method) fitting_loss (ott.neural.methods.monge_gap.MongeGapEstimator property) FixedBarycenter (class in ott.solvers.linear.discrete_barycenter) FixedBarycenterProblem (class in ott.problems.linear.barycenter_problem) fixpoint_iter() (in module ott.math.fixed_point_loop) flattened_b (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) flattened_y (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) flow matching flow_matching_step() (in module ott.neural.methods.flow_matching) FreeBarycenterProblem (class in ott.problems.linear.barycenter_problem) FreeBarycenterState (class in ott.solvers.linear.continuous_barycenter) FreeWassersteinBarycenter (class in ott.solvers.linear.continuous_barycenter) from_graph() (ott.geometry.geodesic.Geodesic class method) (ott.geometry.graph.Graph class method) from_mean_and_cov() (ott.tools.gaussian_mixture.gaussian.Gaussian class method) from_mean_cov_component_weights() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture class method) from_pointcloud() (ott.geometry.low_rank.LRKGeometry class method) from_points_and_assignment_probs() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture class method) from_random() (ott.tools.gaussian_mixture.gaussian.Gaussian class method) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture class method) from_samples() (ott.tools.gaussian_mixture.gaussian.Gaussian class method) from_z() (ott.tools.gaussian_mixture.gaussian.Gaussian method) fu (ott.solvers.linear.sinkhorn.SinkhornState property) G g (ott.neural.methods.expectile_neural_dual.ENOTPotentials attribute) (ott.problems.linear.potentials.DualPotentials attribute) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) (ott.solvers.linear.semidiscrete.HardAssignmentOutput attribute) (ott.solvers.linear.semidiscrete.SemidiscreteOutput attribute) (ott.solvers.linear.semidiscrete.SemidiscreteState attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) g_ema (ott.solvers.linear.semidiscrete.SemidiscreteState attribute) gamma (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) Gangbo-McCann theorem Gaussian (class in ott.tools.gaussian_mixture.gaussian) gaussian_nll() (in module ott.neural.methods.flow_matching) GaussianInitializer (class in ott.initializers.linear.initializers) GaussianMixture (class in ott.tools.gaussian_mixture.gaussian_mixture) GaussianMixturePair (class in ott.tools.gaussian_mixture.gaussian_mixture_pair) GeneralizedKMeansInitializer (class in ott.initializers.linear.initializers_lr) Geodesic (class in ott.geometry.geodesic) geom (ott.solvers.linear.semidiscrete.HardAssignmentOutput property) (ott.solvers.linear.semidiscrete.SemidiscreteOutput property) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) geom_xx (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) geom_xy (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) geom_yy (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) geometries (ott.geometry.grid.Grid property) Geometry (class in ott.geometry.geometry) geoms (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput attribute) get_alpha_schedule() (in module ott.tools.progot) get_bures_geometry() (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair method) get_component() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) get_cost_matrix() (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair method) get_epsilon_schedule() (in module ott.tools.progot) get_fit_model_em_fn() (in module ott.tools.gaussian_mixture.fit_gmm_pair) get_log_component_posterior() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) get_normalized_sinkhorn_coupling() (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair method) get_output() (ott.tools.progot.ProgOTOutput method) get_scores() (ott.tools.conformal.OTCP method) get_sinkhorn() (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair method) get_transport_functions() (ott.problems.linear.linear_problem.LinearProblem method) get_variable() (ott.neural.networks.potentials.BasePotential method) gmm0 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) gmm1 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) grad (ott.neural.networks.layers.conjugate.ConjugateResults attribute) grad_norms (ott.solvers.linear.semidiscrete.SemidiscreteState attribute) gradient() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.icnn.KeyNet method) (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) Graph (class in ott.geometry.graph) Grid (class in ott.geometry.grid) Gromov-Wasserstein distance Gromov-Wasserstein problem GromovWasserstein (class in ott.solvers.quadratic.gromov_wasserstein) GromovWassersteinBarycenter (class in ott.solvers.quadratic.gw_barycenter) ground cost gtol (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) gv (ott.solvers.linear.sinkhorn.SinkhornState property) gw_convergence (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) gw_loss (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) GWBarycenterProblem (class in ott.problems.quadratic.gw_barycenter) GWBarycenterState (class in ott.solvers.quadratic.gw_barycenter) GWLoss (class in ott.problems.quadratic.quadratic_costs) GWOutput (class in ott.solvers.quadratic.gromov_wasserstein) H h() (ott.geometry.costs.EuclideanP method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.RegTICost method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) h1 (ott.problems.quadratic.quadratic_costs.GWLoss attribute) h2 (ott.problems.quadratic.quadratic_costs.GWLoss attribute) h_legendre() (ott.geometry.costs.EuclideanP method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.RegTICost method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) h_transform() (ott.geometry.costs.EuclideanP method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.RegTICost method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) HardAssignmentOutput (class in ott.solvers.linear.semidiscrete) has_nans() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) has_rng() (ott.neural.networks.potentials.BasePotential method) has_variable() (ott.neural.networks.potentials.BasePotential method) histogram (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput attribute) Hungarian algorithm hungarian() (in module ott.tools.unreg) I ICNN (class in ott.neural.networks.icnn) implicit differentiation ImplicitDiff (class in ott.solvers.linear.implicit_differentiation) increase_factor (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) index() (ott.experimental.mmsinkhorn.MMSinkhornOutput method) (ott.neural.networks.layers.conjugate.ConjugateResults method) (ott.problems.quadratic.quadratic_costs.GWLoss method) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState method) (ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn.SinkhornState method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) (ott.solvers.linear.univariate.UnivariateOutput method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState method) (ott.tools.k_means.KMeansOutput method) (ott.tools.progot.ProgOTOutput method) init() (ott.neural.networks.potentials.BasePotential method) init_ema() (in module ott.neural.networks.velocity_field.ema) init_fu() (ott.initializers.linear.initializers.DefaultInitializer method) (ott.initializers.linear.initializers.GaussianInitializer method) (ott.initializers.linear.initializers.SortingInitializer method) (ott.initializers.linear.initializers.SubsampleInitializer method) init_g() (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer method) (ott.initializers.linear.initializers_lr.KMeansInitializer method) (ott.initializers.linear.initializers_lr.LRInitializer method) (ott.initializers.linear.initializers_lr.RandomInitializer method) (ott.initializers.linear.initializers_lr.Rank2Initializer method) init_gv() (ott.initializers.linear.initializers.DefaultInitializer method) (ott.initializers.linear.initializers.GaussianInitializer method) (ott.initializers.linear.initializers.SortingInitializer method) (ott.initializers.linear.initializers.SubsampleInitializer method) init_maps() (ott.solvers.linear.acceleration.AndersonAcceleration method) init_q() (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer method) (ott.initializers.linear.initializers_lr.KMeansInitializer method) (ott.initializers.linear.initializers_lr.LRInitializer method) (ott.initializers.linear.initializers_lr.RandomInitializer method) (ott.initializers.linear.initializers_lr.Rank2Initializer method) init_r() (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer method) (ott.initializers.linear.initializers_lr.KMeansInitializer method) (ott.initializers.linear.initializers_lr.LRInitializer method) (ott.initializers.linear.initializers_lr.RandomInitializer method) (ott.initializers.linear.initializers_lr.Rank2Initializer method) init_state() (ott.experimental.mmsinkhorn.MMSinkhorn method) (ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter method) (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter method) init_transport_mass() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) init_with_output() (ott.neural.networks.potentials.BasePotential method) initialize() (in module ott.tools.gaussian_mixture.fit_gmm) inner_errors (ott.tools.k_means.KMeansOutput attribute) inner_iterations (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) (ott.solvers.linear.acceleration.Momentum attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) input convex neural network interpolate_samples() (in module ott.neural.methods.flow_matching) inv_scale_cost (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) is_assignment (ott.problems.linear.linear_problem.LinearProblem property) is_balanced (ott.problems.linear.linear_problem.LinearProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) is_complement (ott.geometry.regularizers.Quadratic property) is_debiased (ott.tools.progot.ProgOTOutput property) is_entropy_regularized (ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud property) is_equal_size (ott.problems.linear.linear_problem.LinearProblem property) is_factor (ott.geometry.regularizers.Quadratic property) is_fully_orthogonal (ott.geometry.regularizers.Orthogonal property) is_fused (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) is_initializing() (ott.neural.networks.potentials.BasePotential method) is_low_rank (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) (ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter property) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein property) (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter property) (ott.solvers.was_solver.WassersteinSolver property) (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput property) is_mutable_collection() (ott.neural.networks.potentials.BasePotential method) is_neg_dotp (ott.geometry.pointcloud.PointCloud property) is_online (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) is_orthogonal (ott.geometry.regularizers.Quadratic property) is_potential (ott.neural.networks.icnn.ICNN property) (ott.neural.networks.icnn.KeyNet property) (ott.neural.networks.potentials.BasePotential property) (ott.neural.networks.potentials.PotentialMLP property) is_square (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) is_squared_euclidean (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) is_symmetric (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) is_uniform (ott.problems.linear.linear_problem.LinearProblem property) it (ott.solvers.linear.semidiscrete.SemidiscreteOutput attribute) (ott.solvers.linear.semidiscrete.SemidiscreteState attribute) iter_children() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.icnn.KeyNet method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.MLP method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.ema.EMA method) (ott.neural.networks.velocity_field.mlp.MLP method) (ott.neural.networks.velocity_field.unet.UNet method) iter_modules() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.icnn.KeyNet method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.MLP method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.ema.EMA method) (ott.neural.networks.velocity_field.mlp.MLP method) (ott.neural.networks.velocity_field.unet.UNet method) iteration (ott.tools.k_means.KMeansOutput attribute) iterations() (ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter method) K k_means() (in module ott.tools.k_means) Kantorovich problem kernel_matrix (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) kernel_step() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) KeyNet (class in ott.neural.networks.icnn) kl_reg_cost (ott.solvers.linear.sinkhorn.SinkhornOutput property) KMeansInitializer (class in ott.initializers.linear.initializers_lr) KMeansOutput (class in ott.tools.k_means) L L1 (class in ott.geometry.regularizers) lam (ott.geometry.costs.RegTICost property) lazy_init() (ott.neural.networks.potentials.BasePotential method) lbfgs() (in module ott.math) Legendre transform legendre() (in module ott.math) lehmann() (ott.solvers.linear.acceleration.Momentum method) linear_convergence (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) linear_loss (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) linear_outputs (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) linear_state (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) LinearOTDataloader (class in ott.neural.data.ot_dataloader) LinearProblem (class in ott.problems.linear.linear_problem) linesearch_init (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) linesearch_type (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) loc (ott.tools.gaussian_mixture.gaussian.Gaussian property) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) lock_gmm1 (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) log_component_weights() (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) log_prob() (ott.tools.gaussian_mixture.gaussian.Gaussian method) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) logsumexp (in module ott.math.utils) losses (ott.solvers.linear.semidiscrete.SemidiscreteOutput attribute) (ott.solvers.linear.semidiscrete.SemidiscreteState attribute) low-rank optimal transport LRCGeometry (class in ott.geometry.low_rank) LRGromovWasserstein (class in ott.solvers.quadratic.gromov_wasserstein_lr) LRGWOutput (class in ott.solvers.quadratic.gromov_wasserstein_lr) LRInitializer (class in ott.initializers.linear.initializers_lr) LRKGeometry (class in ott.geometry.low_rank) LRSinkhorn (class in ott.solvers.linear.sinkhorn_lr) LRSinkhornOutput (class in ott.solvers.linear.sinkhorn_lr) LRSinkhornState (class in ott.solvers.linear.sinkhorn_lr) lse_step() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) M make_kl_loss() (in module ott.problems.quadratic.quadratic_costs) make_rng() (ott.neural.networks.potentials.BasePotential method) make_square_loss() (in module ott.problems.quadratic.quadratic_costs) marginal() (ott.experimental.mmsinkhorn.MMSinkhornOutput method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) marginal_chi2_error() (ott.solvers.linear.semidiscrete.SemidiscreteOutput method) marginal_dependent_cost() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) marginal_from_potentials() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) marginal_from_scalings() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) marginals (ott.experimental.mmsinkhorn.MMSinkhornOutput property) mass_paired_indices (ott.solvers.linear.univariate.UnivariateOutput attribute) match_linear() (in module ott.solvers.utils) match_quadratic() (in module ott.solvers.utils) matching matrix (ott.solvers.linear.semidiscrete.HardAssignmentOutput property) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) max_iter (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) max_iterations (ott.neural.data.ot_dataloader.LinearOTDataloader attribute) max_linesearch_iter (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS attribute) max_measure_size (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) mean_cost_matrix (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) mean_transport_matrix (ott.solvers.linear.univariate.UnivariateOutput property) median_cost_matrix (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) memory (ott.solvers.linear.acceleration.AndersonAcceleration attribute) MLP (class in ott.neural.networks.potentials) (class in ott.neural.networks.velocity_field.mlp) MMSinkhorn (class in ott.experimental.mmsinkhorn) MMSinkhornOutput (class in ott.experimental.mmsinkhorn) model (ott.tools.conformal.OTCP attribute) module ott ott.experimental ott.experimental.mmsinkhorn ott.geometry ott.geometry.costs ott.initializers ott.initializers.linear ott.initializers.quadratic ott.math ott.neural ott.neural.data ott.neural.methods ott.neural.methods.flow_matching ott.neural.networks ott.neural.networks.layers ott.neural.networks.velocity_field ott.problems ott.problems.linear ott.problems.quadratic ott.solvers ott.solvers.linear ott.solvers.quadratic ott.tools ott.tools.gaussian_mixture ott.utils Momentum (class in ott.solvers.linear.acceleration) Monge map monge_gap() (in module ott.neural.methods.monge_gap) monge_gap_from_samples() (in module ott.neural.methods.monge_gap) MongeGapEstimator (class in ott.neural.methods.monge_gap) moreau_envelope() (ott.geometry.regularizers.L1 method) (ott.geometry.regularizers.Orthogonal method) (ott.geometry.regularizers.PostComposition method) (ott.geometry.regularizers.ProximalOperator method) (ott.geometry.regularizers.Quadratic method) (ott.geometry.regularizers.Regularization method) (ott.geometry.regularizers.SqKOverlap method) (ott.geometry.regularizers.SqL2 method) (ott.geometry.regularizers.STVS method) multimarginal coupling multivariate_cdf_quantile_maps() (in module ott.tools.soft_sort) N n_components (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) n_dimensions (ott.tools.gaussian_mixture.gaussian.Gaussian property) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) n_iters (ott.experimental.mmsinkhorn.MMSinkhornOutput property) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState property) (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput attribute) n_marginals (ott.experimental.mmsinkhorn.MMSinkhornOutput property) name (ott.neural.networks.potentials.BasePotential attribute) ndim (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) ndim_fused (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) NegDotProduct (class in ott.geometry.costs) nonconformity_fn() (ott.tools.conformal.OTCP method) norm() (ott.geometry.costs.Bures method) (ott.geometry.costs.NegDotProduct method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.UnbalancedBures method) norm_error (ott.solvers.linear.sinkhorn.Sinkhorn property) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein property) normalized_entropy (ott.solvers.linear.sinkhorn.SinkhornOutput property) north_west_solver() (in module ott.solvers.linear.univariate) num_iter (ott.neural.networks.layers.conjugate.ConjugateResults attribute) num_iterations (ott.solvers.linear.semidiscrete.SemidiscreteSolver attribute) num_iters (ott.tools.progot.ProgOTOutput property) num_measures (ott.problems.linear.barycenter_problem.FixedBarycenterProblem property) (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) num_per_measure (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) num_steps (ott.tools.progot.ProgOTOutput property) O offset (ott.tools.conformal.OTCP attribute) old_fus (ott.solvers.linear.sinkhorn.SinkhornState attribute) old_mapped_fus (ott.solvers.linear.sinkhorn.SinkhornState attribute) old_transport_mass (ott.solvers.quadratic.gromov_wasserstein.GWOutput attribute) one_iteration() (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) opt_state (ott.neural.networks.potentials.PotentialTrainState attribute) (ott.solvers.linear.semidiscrete.SemidiscreteState attribute) optimal matching problem optimal transport optimizer (ott.solvers.linear.semidiscrete.SemidiscreteSolver attribute) Orthogonal (class in ott.geometry.regularizers) ot_costs (ott.solvers.linear.univariate.UnivariateOutput attribute) ot_prob (ott.solvers.linear.semidiscrete.HardAssignmentOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) OTCP (class in ott.tools.conformal) ott module ott.experimental module ott.experimental.mmsinkhorn module ott.geometry module ott.geometry.costs module ott.initializers module ott.initializers.linear module ott.initializers.quadratic module ott.math module ott.neural module ott.neural.data module ott.neural.methods module ott.neural.methods.flow_matching module ott.neural.networks module ott.neural.networks.layers module ott.neural.networks.velocity_field module ott.problems module ott.problems.linear module ott.problems.quadratic module ott.solvers module ott.solvers.linear module ott.solvers.quadratic module ott.tools module ott.tools.gaussian_mixture module ott.utils module out_shardings (ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader attribute) outer_iterations (ott.experimental.mmsinkhorn.MMSinkhorn property) (ott.solvers.linear.sinkhorn.Sinkhorn property) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein property) output_from_state() (ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter method) (ott.solvers.linear.sinkhorn.Sinkhorn method) (ott.solvers.linear.sinkhorn_lr.LRSinkhorn method) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein method) (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter method) outputs (ott.tools.progot.ProgOTOutput attribute) P paired_indices (ott.solvers.linear.semidiscrete.HardAssignmentOutput attribute) (ott.solvers.linear.univariate.UnivariateOutput attribute) param() (ott.neural.networks.potentials.BasePotential method) params (ott.neural.networks.potentials.PotentialTrainState attribute) parent (ott.neural.networks.potentials.BasePotential attribute) path (ott.neural.networks.potentials.BasePotential property) perturb() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.icnn.KeyNet method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.MLP method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.ema.EMA method) (ott.neural.networks.velocity_field.mlp.MLP method) (ott.neural.networks.velocity_field.unet.UNet method) Plot (class in ott.tools.plot) plot_ot_map() (ott.neural.methods.expectile_neural_dual.ENOTPotentials method) (ott.problems.linear.potentials.DualPotentials method) plot_potential() (ott.neural.methods.expectile_neural_dual.ENOTPotentials method) (ott.problems.linear.potentials.DualPotentials method) PlotMM (class in ott.tools.plot) PNormP (class in ott.geometry.costs) PointCloud (class in ott.geometry.pointcloud) PosDefPotentials (class in ott.neural.networks.layers.posdef) PositiveDense (class in ott.neural.networks.layers.posdef) PostComposition (class in ott.geometry.regularizers) potential_ema (ott.solvers.linear.semidiscrete.SemidiscreteSolver attribute) potential_fn_from_dual_vec() (ott.problems.linear.linear_problem.LinearProblem method) (ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem method) potential_from_scaling() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) potential_gradient_fn (ott.neural.networks.potentials.PotentialTrainState attribute) potential_gradient_fn() (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) potential_value_fn (ott.neural.networks.potentials.PotentialTrainState attribute) potential_value_fn() (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.PotentialMLP method) PotentialMLP (class in ott.neural.networks.potentials) potentials (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn.SinkhornState attribute) (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput attribute) PotentialTrainState (class in ott.neural.networks.potentials) precondition_fun (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) predict() (ott.tools.conformal.OTCP method) prepare_divergences() (ott.geometry.geodesic.Geodesic class method) (ott.geometry.geometry.Geometry class method) (ott.geometry.graph.Graph class method) (ott.geometry.grid.Grid class method) (ott.geometry.low_rank.LRCGeometry class method) (ott.geometry.low_rank.LRKGeometry class method) (ott.geometry.pointcloud.PointCloud class method) primal_cost (ott.solvers.linear.semidiscrete.HardAssignmentOutput property) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein.GWOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) prob (ott.solvers.linear.semidiscrete.SemidiscreteOutput attribute) (ott.solvers.linear.univariate.UnivariateOutput attribute) (ott.tools.progot.ProgOTOutput attribute) ProgOT (class in ott.tools.progot) ProgOTOutput (class in ott.tools.progot) prox() (ott.geometry.regularizers.L1 method) (ott.geometry.regularizers.Orthogonal method) (ott.geometry.regularizers.PostComposition method) (ott.geometry.regularizers.ProximalOperator method) (ott.geometry.regularizers.Quadratic method) (ott.geometry.regularizers.Regularization method) (ott.geometry.regularizers.SqKOverlap method) (ott.geometry.regularizers.SqL2 method) (ott.geometry.regularizers.STVS method) prox_dual() (ott.geometry.regularizers.L1 method) (ott.geometry.regularizers.Orthogonal method) (ott.geometry.regularizers.PostComposition method) (ott.geometry.regularizers.ProximalOperator method) (ott.geometry.regularizers.Quadratic method) (ott.geometry.regularizers.Regularization method) (ott.geometry.regularizers.SqKOverlap method) (ott.geometry.regularizers.SqL2 method) (ott.geometry.regularizers.STVS method) ProximalOperator (class in ott.geometry.regularizers) push-forward put_variable() (ott.neural.networks.potentials.BasePotential method) Q Q (ott.geometry.regularizers.Quadratic property) q (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) quad_loss (ott.problems.quadratic.quadratic_problem.QuadraticProblem property) Quadratic (class in ott.geometry.regularizers) QuadraticInitializer (class in ott.initializers.quadratic.initializers) QuadraticProblem (class in ott.problems.quadratic.quadratic_problem) quantile() (in module ott.tools.soft_sort) quantile_normalization() (in module ott.tools.soft_sort) quantile_solver() (in module ott.solvers.linear.univariate) quantize() (in module ott.tools.soft_sort) R r (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState attribute) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) random_proj_sphere() (in module ott.tools.sliced) RandomInitializer (class in ott.initializers.linear.initializers_lr) rank (ott.geometry.low_rank.LRKGeometry property) (ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer property) (ott.initializers.linear.initializers_lr.KMeansInitializer property) (ott.initializers.linear.initializers_lr.LRInitializer property) (ott.initializers.linear.initializers_lr.RandomInitializer property) (ott.initializers.linear.initializers_lr.Rank2Initializer property) (ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter property) (ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein property) (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter property) (ott.solvers.was_solver.WassersteinSolver property) Rank2Initializer (class in ott.initializers.linear.initializers_lr) ranks() (in module ott.tools.soft_sort) recenter() (ott.solvers.linear.sinkhorn.SinkhornState method) refresh_every (ott.solvers.linear.acceleration.AndersonAcceleration attribute) reg_gw_cost (ott.solvers.quadratic.gromov_wasserstein.GWOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput attribute) reg_ot_cost (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput attribute) reg_ot_cost() (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) RegTICost (class in ott.geometry.costs) Regularization (class in ott.geometry.regularizers) regularizer (ott.neural.methods.monge_gap.MongeGapEstimator property) relative_epsilon (ott.neural.data.ot_dataloader.LinearOTDataloader attribute) replace (ott.neural.data.ot_dataloader.LinearOTDataloader attribute) replace() (ott.neural.networks.potentials.PotentialTrainState method) (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) return_indices (ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader attribute) rho (ott.geometry.costs.RegTICost property) (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) ridge_identity (ott.solvers.linear.acceleration.AndersonAcceleration attribute) rng (ott.neural.data.ot_dataloader.LinearOTDataloader attribute) (ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader attribute) S sample() (ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud method) (ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem method) (ott.solvers.linear.semidiscrete.SemidiscreteOutput method) (ott.tools.gaussian_mixture.gaussian.Gaussian method) (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture method) sample_conditional() (in module ott.solvers.utils) sample_joint() (in module ott.solvers.utils) sampler (ott.tools.conformal.OTCP attribute) scale (ott.tools.conformal.OTCP attribute) (ott.tools.gaussian_mixture.gaussian.Gaussian property) scale_params (ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture property) scaling_from_potential() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) scalings (ott.solvers.linear.sinkhorn.SinkhornOutput property) scope (ott.neural.networks.potentials.BasePotential attribute) sd_out (ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader attribute) segment_point_cloud() (in module ott.geometry.segment) segment_sinkhorn() (in module ott.tools.segment_sinkhorn) segment_sinkhorn_divergence() (in module ott.tools.sinkhorn_divergence) segmented_y_b (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) segmented_y_fused (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) semidiscrete problem SemidiscreteDataloader (class in ott.neural.data.semidiscrete_dataloader) SemidiscreteLinearProblem (class in ott.problems.linear.semidiscrete_linear_problem) SemidiscreteOutput (class in ott.solvers.linear.semidiscrete) SemidiscretePointCloud (class in ott.geometry.semidiscrete_pointcloud) SemidiscreteSolver (class in ott.solvers.linear.semidiscrete) SemidiscreteState (class in ott.solvers.linear.semidiscrete) set() (ott.experimental.mmsinkhorn.MMSinkhornOutput method) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn.SinkhornState method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) (ott.solvers.quadratic.gromov_wasserstein.GWOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState method) set_cost() (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) set_scale_cost() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) setup() (ott.neural.methods.monge_gap.MongeGapEstimator method) (ott.neural.methods.neuraldual.W2NeuralDual method) (ott.neural.networks.potentials.BasePotential method) shape (ott.experimental.mmsinkhorn.MMSinkhornOutput property) (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) (ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud property) shardings (ott.neural.data.ot_dataloader.LinearOTDataloader attribute) sinkdiv() (in module ott.tools.sinkhorn_divergence) Sinkhorn (class in ott.solvers.linear.sinkhorn) Sinkhorn algorithm Sinkhorn divergence sinkhorn_divergence() (in module ott.tools.sinkhorn_divergence) sinkhorn_output (ott.tools.conformal.OTCP attribute) SinkhornBarycenterOutput (class in ott.solvers.linear.discrete_barycenter) SinkhornDivergenceOutput (class in ott.tools.sinkhorn_divergence) SinkhornOutput (class in ott.solvers.linear.sinkhorn) SinkhornState (class in ott.solvers.linear.sinkhorn) sliced_wasserstein() (in module ott.tools.sliced) sobol_ball_sampler() (in module ott.tools.conformal) SoftDTW (class in ott.geometry.costs) solution_error() (ott.solvers.linear.sinkhorn.SinkhornState method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornState method) solve() (in module ott.solvers.linear) (in module ott.solvers.quadratic) (ott.neural.networks.layers.conjugate.FenchelConjugateLBFGS method) (ott.neural.networks.layers.conjugate.FenchelConjugateSolver method) (ott.solvers.linear.implicit_differentiation.ImplicitDiff method) solve_jax_cg() (in module ott.solvers.linear.implicit_differentiation) solve_lineax() (in module ott.solvers.linear.lineax_implicit) solve_semidiscrete() (in module ott.solvers.linear) solve_univariate() (in module ott.solvers.linear) solver (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) solver_kwargs (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) sort() (in module ott.tools.soft_sort) sort_with() (in module ott.tools.soft_sort) SortingInitializer (class in ott.initializers.linear.initializers) sow() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.icnn.KeyNet method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.BasePotential method) (ott.neural.networks.potentials.MLP method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.ema.EMA method) (ott.neural.networks.velocity_field.mlp.MLP method) (ott.neural.networks.velocity_field.unet.UNet method) SqEuclidean (class in ott.geometry.costs) SqKOverlap (class in ott.geometry.regularizers) SqL2 (class in ott.geometry.regularizers) SqPNorm (class in ott.geometry.costs) start (ott.solvers.linear.acceleration.Momentum attribute) std_cost_matrix (ott.geometry.geodesic.Geodesic property) (ott.geometry.geometry.Geometry property) (ott.geometry.graph.Graph property) (ott.geometry.grid.Grid property) (ott.geometry.low_rank.LRCGeometry property) (ott.geometry.low_rank.LRKGeometry property) (ott.geometry.pointcloud.PointCloud property) step (ott.neural.networks.potentials.PotentialTrainState attribute) step() (ott.solvers.linear.semidiscrete.SemidiscreteSolver method) STVS (class in ott.geometry.regularizers) SubsampleInitializer (class in ott.initializers.linear.initializers) subset() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) subset_size (ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader attribute) subset_size_threshold (ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader attribute) symmetric (ott.solvers.linear.implicit_differentiation.ImplicitDiff attribute) T target_measure (ott.tools.conformal.OTCP property) tau (ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair property) tensor (ott.experimental.mmsinkhorn.MMSinkhornOutput property) third_lower_bound() (in module ott.solvers.quadratic.lower_bound) threshold (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) (ott.neural.data.ot_dataloader.LinearOTDataloader attribute) (ott.solvers.linear.semidiscrete.SemidiscreteSolver attribute) (ott.solvers.linear.sinkhorn.SinkhornOutput attribute) TICost (class in ott.geometry.costs) to_dataloader() (ott.solvers.linear.semidiscrete.SemidiscreteOutput method) to_dual_potentials() (ott.neural.methods.expectile_neural_dual.ExpectileNeuralDual method) (ott.neural.methods.neuraldual.W2NeuralDual method) (ott.solvers.linear.semidiscrete.SemidiscreteOutput method) (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput method) to_low_rank() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) to_LRCGeometry() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) to_z() (ott.tools.gaussian_mixture.gaussian.Gaussian method) topk_mask() (in module ott.tools.soft_sort) tqdm_progress_fn() (in module ott.utils) train() (ott.neural.networks.icnn.ICNN method) (ott.neural.networks.icnn.KeyNet method) (ott.neural.networks.layers.posdef.PosDefPotentials method) (ott.neural.networks.layers.posdef.PositiveDense method) (ott.neural.networks.potentials.MLP method) (ott.neural.networks.potentials.PotentialMLP method) (ott.neural.networks.velocity_field.ema.EMA method) (ott.neural.networks.velocity_field.mlp.MLP method) (ott.neural.networks.velocity_field.unet.UNet method) train_fn() (ott.neural.methods.expectile_neural_dual.ExpectileNeuralDual method) train_map_estimator() (ott.neural.methods.monge_gap.MongeGapEstimator method) train_neuraldual_alternating() (ott.neural.methods.neuraldual.W2NeuralDual method) train_neuraldual_parallel() (ott.neural.methods.neuraldual.W2NeuralDual method) transport map transport plan transport() (ott.neural.methods.expectile_neural_dual.ENOTPotentials method) (ott.problems.linear.potentials.DualPotentials method) (ott.tools.gaussian_mixture.gaussian.Gaussian method) (ott.tools.progot.ProgOTOutput method) transport_animation() (in module ott.tools.plot) transport_cost_at_geom() (ott.solvers.linear.sinkhorn.SinkhornOutput method) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput method) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput method) transport_from_potentials() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) transport_from_scalings() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) transport_map() (ott.geometry.costs.EuclideanP method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.RegTICost method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) transport_mass (ott.experimental.mmsinkhorn.MMSinkhornOutput property) (ott.solvers.linear.sinkhorn.SinkhornOutput property) (ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput property) (ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput property) transport_matrices (ott.solvers.linear.univariate.UnivariateOutput property) transportation polytope twist condition twist_operator() (ott.geometry.costs.Arccos method) (ott.geometry.costs.Bures method) (ott.geometry.costs.Cosine method) (ott.geometry.costs.CostFn method) (ott.geometry.costs.Euclidean method) (ott.geometry.costs.EuclideanP method) (ott.geometry.costs.NegDotProduct method) (ott.geometry.costs.PNormP method) (ott.geometry.costs.RegTICost method) (ott.geometry.costs.SoftDTW method) (ott.geometry.costs.SqEuclidean method) (ott.geometry.costs.SqPNorm method) (ott.geometry.costs.TICost method) (ott.geometry.costs.UnbalancedBures method) (ott.geometry.distrib_costs.UnivariateWasserstein method) tx (ott.neural.networks.potentials.PotentialTrainState attribute) U unbalanced unbalanced_dykstra_kernel() (in module ott.solvers.linear.lr_utils) unbalanced_dykstra_lse() (in module ott.solvers.linear.lr_utils) UnbalancedBures (class in ott.geometry.costs) unbind() (ott.neural.networks.potentials.BasePotential method) UNet (class in ott.neural.networks.velocity_field.unet) uniform_sampler() (in module ott.solvers.utils) uniform_solver() (in module ott.solvers.linear.univariate) UnivariateOutput (class in ott.solvers.linear.univariate) UnivariateWasserstein (class in ott.geometry.distrib_costs) unrolling update() (ott.solvers.linear.acceleration.AndersonAcceleration method) (ott.solvers.linear.continuous_barycenter.FreeBarycenterState method) (ott.tools.plot.Plot method) (ott.tools.plot.PlotMM method) update_barycenter() (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem method) update_ema() (in module ott.neural.networks.velocity_field.ema) update_features() (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem method) update_history() (ott.solvers.linear.acceleration.AndersonAcceleration method) update_linearization() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) update_lr_geom() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) update_lr_linearization() (ott.problems.quadratic.quadratic_problem.QuadraticProblem method) update_potential() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) update_scaling() (ott.geometry.geodesic.Geodesic method) (ott.geometry.geometry.Geometry method) (ott.geometry.graph.Graph method) (ott.geometry.grid.Grid method) (ott.geometry.low_rank.LRCGeometry method) (ott.geometry.low_rank.LRKGeometry method) (ott.geometry.pointcloud.PointCloud method) update_state() (ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter method) V val (ott.neural.networks.layers.conjugate.ConjugateResults attribute) value (ott.solvers.linear.acceleration.Momentum attribute) variable() (ott.neural.networks.potentials.BasePotential method) variables (ott.neural.networks.potentials.BasePotential property) velocity_from_brenier_potential() (in module ott.math) W w2_dist() (ott.tools.gaussian_mixture.gaussian.Gaussian method) W2NeuralDual (class in ott.neural.methods.neuraldual) wassdis_p() (in module ott.tools.unreg) Wasserstein barycenter Wasserstein distance WassersteinSolver (class in ott.solvers.was_solver) weight() (ott.solvers.linear.acceleration.Momentum method) weights (ott.problems.linear.barycenter_problem.FixedBarycenterProblem property) (ott.problems.linear.barycenter_problem.FreeBarycenterProblem property) (ott.problems.quadratic.gw_barycenter.GWBarycenterProblem property) X x (ott.solvers.linear.continuous_barycenter.FreeBarycenterState attribute) (ott.solvers.quadratic.gw_barycenter.GWBarycenterState attribute) x_s (ott.experimental.mmsinkhorn.MMSinkhornOutput attribute) xs (ott.tools.progot.ProgOTOutput attribute)
