Skip to main content
Back to top
Ctrl
+
K
Search
Ctrl
+
K
Examples
Getting Started
Tutorials
Geometry
Point Clouds
Grid Geometry
Linear OT
Focus on Sinkhorn
OTT & POT
Sinkhorn divergence gradient flows
Low-rank Sinkhorn
Sinkhorn Divergence Hessians
MBO Sparse Maps
Multimarginal OT
Progressive Entropic Optimal Transport
Unbalanced Optimal Transport
Quadratic OT
Gromov-Wasserstein
Low-rank GW
GW for Multi-omics
Neural OT
Neural Dual Solver
ICNN Initialization
Monge gap
Expectile-Regularized OT
Meta OT
Barycenter
Sinkhorn Barycenters
GMM Barycenters
Fitting Coupled GMMs
Miscellaneous
Tracking progress of
ott.solvers
Soft Sorting
Single-cell Genomics
API
ott.geometry
ott.geometry.geometry.Geometry
ott.geometry.geometry.Geometry.apply_cost
ott.geometry.geometry.Geometry.apply_kernel
ott.geometry.geometry.Geometry.apply_lse_kernel
ott.geometry.geometry.Geometry.apply_square_cost
ott.geometry.geometry.Geometry.apply_transport_from_potentials
ott.geometry.geometry.Geometry.apply_transport_from_scalings
ott.geometry.geometry.Geometry.copy_epsilon
ott.geometry.geometry.Geometry.marginal_from_potentials
ott.geometry.geometry.Geometry.marginal_from_scalings
ott.geometry.geometry.Geometry.potential_from_scaling
ott.geometry.geometry.Geometry.prepare_divergences
ott.geometry.geometry.Geometry.scaling_from_potential
ott.geometry.geometry.Geometry.set_scale_cost
ott.geometry.geometry.Geometry.subset
ott.geometry.geometry.Geometry.to_LRCGeometry
ott.geometry.geometry.Geometry.transport_from_potentials
ott.geometry.geometry.Geometry.transport_from_scalings
ott.geometry.geometry.Geometry.update_potential
ott.geometry.geometry.Geometry.update_scaling
ott.geometry.geometry.Geometry.can_LRC
ott.geometry.geometry.Geometry.cost_matrix
ott.geometry.geometry.Geometry.cost_rank
ott.geometry.geometry.Geometry.dtype
ott.geometry.geometry.Geometry.epsilon
ott.geometry.geometry.Geometry.epsilon_scheduler
ott.geometry.geometry.Geometry.inv_scale_cost
ott.geometry.geometry.Geometry.is_online
ott.geometry.geometry.Geometry.is_squared_euclidean
ott.geometry.geometry.Geometry.is_symmetric
ott.geometry.geometry.Geometry.kernel_matrix
ott.geometry.geometry.Geometry.mean_cost_matrix
ott.geometry.geometry.Geometry.median_cost_matrix
ott.geometry.geometry.Geometry.shape
ott.geometry.geometry.Geometry.std_cost_matrix
ott.geometry.pointcloud.PointCloud
ott.geometry.pointcloud.PointCloud.apply_cost
ott.geometry.pointcloud.PointCloud.apply_kernel
ott.geometry.pointcloud.PointCloud.apply_lse_kernel
ott.geometry.pointcloud.PointCloud.apply_square_cost
ott.geometry.pointcloud.PointCloud.apply_transport_from_potentials
ott.geometry.pointcloud.PointCloud.apply_transport_from_scalings
ott.geometry.pointcloud.PointCloud.barycenter
ott.geometry.pointcloud.PointCloud.copy_epsilon
ott.geometry.pointcloud.PointCloud.marginal_from_potentials
ott.geometry.pointcloud.PointCloud.marginal_from_scalings
ott.geometry.pointcloud.PointCloud.potential_from_scaling
ott.geometry.pointcloud.PointCloud.prepare_divergences
ott.geometry.pointcloud.PointCloud.scaling_from_potential
ott.geometry.pointcloud.PointCloud.set_scale_cost
ott.geometry.pointcloud.PointCloud.subset
ott.geometry.pointcloud.PointCloud.to_LRCGeometry
ott.geometry.pointcloud.PointCloud.transport_from_potentials
ott.geometry.pointcloud.PointCloud.transport_from_scalings
ott.geometry.pointcloud.PointCloud.update_potential
ott.geometry.pointcloud.PointCloud.update_scaling
ott.geometry.pointcloud.PointCloud.batch_size
ott.geometry.pointcloud.PointCloud.can_LRC
ott.geometry.pointcloud.PointCloud.cost_matrix
ott.geometry.pointcloud.PointCloud.cost_rank
ott.geometry.pointcloud.PointCloud.dtype
ott.geometry.pointcloud.PointCloud.epsilon
ott.geometry.pointcloud.PointCloud.epsilon_scheduler
ott.geometry.pointcloud.PointCloud.inv_scale_cost
ott.geometry.pointcloud.PointCloud.is_online
ott.geometry.pointcloud.PointCloud.is_squared_euclidean
ott.geometry.pointcloud.PointCloud.is_symmetric
ott.geometry.pointcloud.PointCloud.kernel_matrix
ott.geometry.pointcloud.PointCloud.mean_cost_matrix
ott.geometry.pointcloud.PointCloud.median_cost_matrix
ott.geometry.pointcloud.PointCloud.shape
ott.geometry.pointcloud.PointCloud.std_cost_matrix
ott.geometry.grid.Grid
ott.geometry.grid.Grid.apply_cost
ott.geometry.grid.Grid.apply_kernel
ott.geometry.grid.Grid.apply_lse_kernel
ott.geometry.grid.Grid.apply_square_cost
ott.geometry.grid.Grid.apply_transport_from_potentials
ott.geometry.grid.Grid.apply_transport_from_scalings
ott.geometry.grid.Grid.copy_epsilon
ott.geometry.grid.Grid.marginal_from_potentials
ott.geometry.grid.Grid.marginal_from_scalings
ott.geometry.grid.Grid.potential_from_scaling
ott.geometry.grid.Grid.prepare_divergences
ott.geometry.grid.Grid.scaling_from_potential
ott.geometry.grid.Grid.set_scale_cost
ott.geometry.grid.Grid.subset
ott.geometry.grid.Grid.to_LRCGeometry
ott.geometry.grid.Grid.transport_from_potentials
ott.geometry.grid.Grid.transport_from_scalings
ott.geometry.grid.Grid.update_potential
ott.geometry.grid.Grid.update_scaling
ott.geometry.grid.Grid.can_LRC
ott.geometry.grid.Grid.cost_matrix
ott.geometry.grid.Grid.cost_rank
ott.geometry.grid.Grid.dtype
ott.geometry.grid.Grid.epsilon
ott.geometry.grid.Grid.epsilon_scheduler
ott.geometry.grid.Grid.geometries
ott.geometry.grid.Grid.inv_scale_cost
ott.geometry.grid.Grid.is_online
ott.geometry.grid.Grid.is_squared_euclidean
ott.geometry.grid.Grid.is_symmetric
ott.geometry.grid.Grid.kernel_matrix
ott.geometry.grid.Grid.mean_cost_matrix
ott.geometry.grid.Grid.median_cost_matrix
ott.geometry.grid.Grid.shape
ott.geometry.grid.Grid.std_cost_matrix
ott.geometry.graph.Graph
ott.geometry.graph.Graph.apply_cost
ott.geometry.graph.Graph.apply_kernel
ott.geometry.graph.Graph.apply_lse_kernel
ott.geometry.graph.Graph.apply_square_cost
ott.geometry.graph.Graph.apply_transport_from_potentials
ott.geometry.graph.Graph.apply_transport_from_scalings
ott.geometry.graph.Graph.copy_epsilon
ott.geometry.graph.Graph.from_graph
ott.geometry.graph.Graph.marginal_from_potentials
ott.geometry.graph.Graph.marginal_from_scalings
ott.geometry.graph.Graph.potential_from_scaling
ott.geometry.graph.Graph.prepare_divergences
ott.geometry.graph.Graph.scaling_from_potential
ott.geometry.graph.Graph.set_scale_cost
ott.geometry.graph.Graph.subset
ott.geometry.graph.Graph.to_LRCGeometry
ott.geometry.graph.Graph.transport_from_potentials
ott.geometry.graph.Graph.transport_from_scalings
ott.geometry.graph.Graph.update_potential
ott.geometry.graph.Graph.update_scaling
ott.geometry.graph.Graph.can_LRC
ott.geometry.graph.Graph.cost_matrix
ott.geometry.graph.Graph.cost_rank
ott.geometry.graph.Graph.dtype
ott.geometry.graph.Graph.epsilon
ott.geometry.graph.Graph.epsilon_scheduler
ott.geometry.graph.Graph.inv_scale_cost
ott.geometry.graph.Graph.is_online
ott.geometry.graph.Graph.is_squared_euclidean
ott.geometry.graph.Graph.is_symmetric
ott.geometry.graph.Graph.kernel_matrix
ott.geometry.graph.Graph.mean_cost_matrix
ott.geometry.graph.Graph.median_cost_matrix
ott.geometry.graph.Graph.shape
ott.geometry.graph.Graph.std_cost_matrix
ott.geometry.geodesic.Geodesic
ott.geometry.geodesic.Geodesic.apply_cost
ott.geometry.geodesic.Geodesic.apply_kernel
ott.geometry.geodesic.Geodesic.apply_lse_kernel
ott.geometry.geodesic.Geodesic.apply_square_cost
ott.geometry.geodesic.Geodesic.apply_transport_from_potentials
ott.geometry.geodesic.Geodesic.apply_transport_from_scalings
ott.geometry.geodesic.Geodesic.copy_epsilon
ott.geometry.geodesic.Geodesic.from_graph
ott.geometry.geodesic.Geodesic.marginal_from_potentials
ott.geometry.geodesic.Geodesic.marginal_from_scalings
ott.geometry.geodesic.Geodesic.potential_from_scaling
ott.geometry.geodesic.Geodesic.prepare_divergences
ott.geometry.geodesic.Geodesic.scaling_from_potential
ott.geometry.geodesic.Geodesic.set_scale_cost
ott.geometry.geodesic.Geodesic.subset
ott.geometry.geodesic.Geodesic.to_LRCGeometry
ott.geometry.geodesic.Geodesic.transport_from_potentials
ott.geometry.geodesic.Geodesic.transport_from_scalings
ott.geometry.geodesic.Geodesic.update_potential
ott.geometry.geodesic.Geodesic.update_scaling
ott.geometry.geodesic.Geodesic.can_LRC
ott.geometry.geodesic.Geodesic.cost_matrix
ott.geometry.geodesic.Geodesic.cost_rank
ott.geometry.geodesic.Geodesic.dtype
ott.geometry.geodesic.Geodesic.epsilon
ott.geometry.geodesic.Geodesic.epsilon_scheduler
ott.geometry.geodesic.Geodesic.inv_scale_cost
ott.geometry.geodesic.Geodesic.is_online
ott.geometry.geodesic.Geodesic.is_squared_euclidean
ott.geometry.geodesic.Geodesic.is_symmetric
ott.geometry.geodesic.Geodesic.kernel_matrix
ott.geometry.geodesic.Geodesic.mean_cost_matrix
ott.geometry.geodesic.Geodesic.median_cost_matrix
ott.geometry.geodesic.Geodesic.shape
ott.geometry.geodesic.Geodesic.std_cost_matrix
ott.geometry.low_rank.LRCGeometry
ott.geometry.low_rank.LRCGeometry.apply_cost
ott.geometry.low_rank.LRCGeometry.apply_kernel
ott.geometry.low_rank.LRCGeometry.apply_lse_kernel
ott.geometry.low_rank.LRCGeometry.apply_square_cost
ott.geometry.low_rank.LRCGeometry.apply_transport_from_potentials
ott.geometry.low_rank.LRCGeometry.apply_transport_from_scalings
ott.geometry.low_rank.LRCGeometry.copy_epsilon
ott.geometry.low_rank.LRCGeometry.marginal_from_potentials
ott.geometry.low_rank.LRCGeometry.marginal_from_scalings
ott.geometry.low_rank.LRCGeometry.potential_from_scaling
ott.geometry.low_rank.LRCGeometry.prepare_divergences
ott.geometry.low_rank.LRCGeometry.scaling_from_potential
ott.geometry.low_rank.LRCGeometry.set_scale_cost
ott.geometry.low_rank.LRCGeometry.subset
ott.geometry.low_rank.LRCGeometry.to_LRCGeometry
ott.geometry.low_rank.LRCGeometry.transport_from_potentials
ott.geometry.low_rank.LRCGeometry.transport_from_scalings
ott.geometry.low_rank.LRCGeometry.update_potential
ott.geometry.low_rank.LRCGeometry.update_scaling
ott.geometry.low_rank.LRCGeometry.bias
ott.geometry.low_rank.LRCGeometry.can_LRC
ott.geometry.low_rank.LRCGeometry.cost_1
ott.geometry.low_rank.LRCGeometry.cost_2
ott.geometry.low_rank.LRCGeometry.cost_matrix
ott.geometry.low_rank.LRCGeometry.cost_rank
ott.geometry.low_rank.LRCGeometry.dtype
ott.geometry.low_rank.LRCGeometry.epsilon
ott.geometry.low_rank.LRCGeometry.epsilon_scheduler
ott.geometry.low_rank.LRCGeometry.inv_scale_cost
ott.geometry.low_rank.LRCGeometry.is_online
ott.geometry.low_rank.LRCGeometry.is_squared_euclidean
ott.geometry.low_rank.LRCGeometry.is_symmetric
ott.geometry.low_rank.LRCGeometry.kernel_matrix
ott.geometry.low_rank.LRCGeometry.mean_cost_matrix
ott.geometry.low_rank.LRCGeometry.median_cost_matrix
ott.geometry.low_rank.LRCGeometry.shape
ott.geometry.low_rank.LRCGeometry.std_cost_matrix
ott.geometry.low_rank.LRKGeometry
ott.geometry.low_rank.LRKGeometry.apply_cost
ott.geometry.low_rank.LRKGeometry.apply_kernel
ott.geometry.low_rank.LRKGeometry.apply_lse_kernel
ott.geometry.low_rank.LRKGeometry.apply_square_cost
ott.geometry.low_rank.LRKGeometry.apply_transport_from_potentials
ott.geometry.low_rank.LRKGeometry.apply_transport_from_scalings
ott.geometry.low_rank.LRKGeometry.copy_epsilon
ott.geometry.low_rank.LRKGeometry.from_pointcloud
ott.geometry.low_rank.LRKGeometry.marginal_from_potentials
ott.geometry.low_rank.LRKGeometry.marginal_from_scalings
ott.geometry.low_rank.LRKGeometry.potential_from_scaling
ott.geometry.low_rank.LRKGeometry.prepare_divergences
ott.geometry.low_rank.LRKGeometry.scaling_from_potential
ott.geometry.low_rank.LRKGeometry.set_scale_cost
ott.geometry.low_rank.LRKGeometry.subset
ott.geometry.low_rank.LRKGeometry.to_LRCGeometry
ott.geometry.low_rank.LRKGeometry.transport_from_potentials
ott.geometry.low_rank.LRKGeometry.transport_from_scalings
ott.geometry.low_rank.LRKGeometry.update_potential
ott.geometry.low_rank.LRKGeometry.update_scaling
ott.geometry.low_rank.LRKGeometry.can_LRC
ott.geometry.low_rank.LRKGeometry.cost_matrix
ott.geometry.low_rank.LRKGeometry.cost_rank
ott.geometry.low_rank.LRKGeometry.dtype
ott.geometry.low_rank.LRKGeometry.epsilon
ott.geometry.low_rank.LRKGeometry.epsilon_scheduler
ott.geometry.low_rank.LRKGeometry.inv_scale_cost
ott.geometry.low_rank.LRKGeometry.is_online
ott.geometry.low_rank.LRKGeometry.is_squared_euclidean
ott.geometry.low_rank.LRKGeometry.is_symmetric
ott.geometry.low_rank.LRKGeometry.kernel_matrix
ott.geometry.low_rank.LRKGeometry.mean_cost_matrix
ott.geometry.low_rank.LRKGeometry.median_cost_matrix
ott.geometry.low_rank.LRKGeometry.rank
ott.geometry.low_rank.LRKGeometry.shape
ott.geometry.low_rank.LRKGeometry.std_cost_matrix
ott.geometry.epsilon_scheduler.Epsilon
ott.geometry.epsilon_scheduler.DEFAULT_EPSILON_SCALE
ott.geometry.costs.CostFn
ott.geometry.costs.CostFn.all_pairs
ott.geometry.costs.CostFn.barycenter
ott.geometry.costs.CostFn.twist_operator
ott.geometry.costs.TICost
ott.geometry.costs.TICost.all_pairs
ott.geometry.costs.TICost.barycenter
ott.geometry.costs.TICost.h
ott.geometry.costs.TICost.h_legendre
ott.geometry.costs.TICost.h_transform
ott.geometry.costs.TICost.transport_map
ott.geometry.costs.TICost.twist_operator
ott.geometry.costs.SqPNorm
ott.geometry.costs.SqPNorm.all_pairs
ott.geometry.costs.SqPNorm.barycenter
ott.geometry.costs.SqPNorm.h
ott.geometry.costs.SqPNorm.h_legendre
ott.geometry.costs.SqPNorm.h_transform
ott.geometry.costs.SqPNorm.transport_map
ott.geometry.costs.SqPNorm.twist_operator
ott.geometry.costs.PNormP
ott.geometry.costs.PNormP.all_pairs
ott.geometry.costs.PNormP.barycenter
ott.geometry.costs.PNormP.h
ott.geometry.costs.PNormP.h_legendre
ott.geometry.costs.PNormP.h_transform
ott.geometry.costs.PNormP.transport_map
ott.geometry.costs.PNormP.twist_operator
ott.geometry.costs.SqEuclidean
ott.geometry.costs.SqEuclidean.all_pairs
ott.geometry.costs.SqEuclidean.barycenter
ott.geometry.costs.SqEuclidean.h
ott.geometry.costs.SqEuclidean.h_legendre
ott.geometry.costs.SqEuclidean.h_transform
ott.geometry.costs.SqEuclidean.norm
ott.geometry.costs.SqEuclidean.transport_map
ott.geometry.costs.SqEuclidean.twist_operator
ott.geometry.costs.RegTICost
ott.geometry.costs.RegTICost.all_pairs
ott.geometry.costs.RegTICost.barycenter
ott.geometry.costs.RegTICost.h
ott.geometry.costs.RegTICost.h_legendre
ott.geometry.costs.RegTICost.h_transform
ott.geometry.costs.RegTICost.transport_map
ott.geometry.costs.RegTICost.twist_operator
ott.geometry.costs.RegTICost.lam
ott.geometry.costs.RegTICost.rho
ott.geometry.costs.Euclidean
ott.geometry.costs.Euclidean.all_pairs
ott.geometry.costs.Euclidean.barycenter
ott.geometry.costs.Euclidean.twist_operator
ott.geometry.costs.EuclideanP
ott.geometry.costs.EuclideanP.all_pairs
ott.geometry.costs.EuclideanP.barycenter
ott.geometry.costs.EuclideanP.h
ott.geometry.costs.EuclideanP.h_legendre
ott.geometry.costs.EuclideanP.h_transform
ott.geometry.costs.EuclideanP.transport_map
ott.geometry.costs.EuclideanP.twist_operator
ott.geometry.costs.Cosine
ott.geometry.costs.Cosine.all_pairs
ott.geometry.costs.Cosine.barycenter
ott.geometry.costs.Cosine.twist_operator
ott.geometry.costs.Arccos
ott.geometry.costs.Arccos.all_pairs
ott.geometry.costs.Arccos.barycenter
ott.geometry.costs.Arccos.twist_operator
ott.geometry.costs.Bures
ott.geometry.costs.Bures.all_pairs
ott.geometry.costs.Bures.barycenter
ott.geometry.costs.Bures.covariance_fixpoint_iter
ott.geometry.costs.Bures.norm
ott.geometry.costs.Bures.twist_operator
ott.geometry.costs.UnbalancedBures
ott.geometry.costs.UnbalancedBures.all_pairs
ott.geometry.costs.UnbalancedBures.barycenter
ott.geometry.costs.UnbalancedBures.norm
ott.geometry.costs.UnbalancedBures.twist_operator
ott.geometry.costs.SoftDTW
ott.geometry.costs.SoftDTW.all_pairs
ott.geometry.costs.SoftDTW.barycenter
ott.geometry.costs.SoftDTW.twist_operator
ott.geometry.distrib_costs.UnivariateWasserstein
ott.geometry.distrib_costs.UnivariateWasserstein.all_pairs
ott.geometry.distrib_costs.UnivariateWasserstein.barycenter
ott.geometry.distrib_costs.UnivariateWasserstein.twist_operator
ott.geometry.regularizers.ProximalOperator
ott.geometry.regularizers.ProximalOperator.moreau_envelope
ott.geometry.regularizers.ProximalOperator.prox
ott.geometry.regularizers.ProximalOperator.prox_dual
ott.geometry.regularizers.PostComposition
ott.geometry.regularizers.PostComposition.moreau_envelope
ott.geometry.regularizers.PostComposition.prox
ott.geometry.regularizers.PostComposition.prox_dual
ott.geometry.regularizers.Regularization
ott.geometry.regularizers.Regularization.moreau_envelope
ott.geometry.regularizers.Regularization.prox
ott.geometry.regularizers.Regularization.prox_dual
ott.geometry.regularizers.Orthogonal
ott.geometry.regularizers.Orthogonal.moreau_envelope
ott.geometry.regularizers.Orthogonal.prox
ott.geometry.regularizers.Orthogonal.prox_dual
ott.geometry.regularizers.Orthogonal.is_fully_orthogonal
ott.geometry.regularizers.Quadratic
ott.geometry.regularizers.Quadratic.moreau_envelope
ott.geometry.regularizers.Quadratic.prox
ott.geometry.regularizers.Quadratic.prox_dual
ott.geometry.regularizers.Quadratic.A_comp
ott.geometry.regularizers.Quadratic.Q
ott.geometry.regularizers.Quadratic.is_complement
ott.geometry.regularizers.Quadratic.is_factor
ott.geometry.regularizers.Quadratic.is_orthogonal
ott.geometry.regularizers.L1
ott.geometry.regularizers.L1.moreau_envelope
ott.geometry.regularizers.L1.prox
ott.geometry.regularizers.L1.prox_dual
ott.geometry.regularizers.SqL2
ott.geometry.regularizers.SqL2.moreau_envelope
ott.geometry.regularizers.SqL2.prox
ott.geometry.regularizers.SqL2.prox_dual
ott.geometry.regularizers.STVS
ott.geometry.regularizers.STVS.moreau_envelope
ott.geometry.regularizers.STVS.prox
ott.geometry.regularizers.STVS.prox_dual
ott.geometry.regularizers.SqKOverlap
ott.geometry.regularizers.SqKOverlap.moreau_envelope
ott.geometry.regularizers.SqKOverlap.prox
ott.geometry.regularizers.SqKOverlap.prox_dual
ott.geometry.segment.segment_point_cloud
ott.problems
ott.problems.linear
ott.problems.linear.linear_problem.LinearProblem
ott.problems.linear.linear_problem.LinearProblem.get_transport_functions
ott.problems.linear.linear_problem.LinearProblem.a
ott.problems.linear.linear_problem.LinearProblem.b
ott.problems.linear.linear_problem.LinearProblem.dtype
ott.problems.linear.linear_problem.LinearProblem.epsilon
ott.problems.linear.linear_problem.LinearProblem.is_balanced
ott.problems.linear.linear_problem.LinearProblem.is_equal_size
ott.problems.linear.linear_problem.LinearProblem.is_uniform
ott.problems.linear.barycenter_problem.FixedBarycenterProblem
ott.problems.linear.barycenter_problem.FixedBarycenterProblem.num_measures
ott.problems.linear.barycenter_problem.FixedBarycenterProblem.weights
ott.problems.linear.barycenter_problem.FreeBarycenterProblem
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.flattened_b
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.flattened_y
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.max_measure_size
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.ndim
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.num_measures
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.segmented_y_b
ott.problems.linear.barycenter_problem.FreeBarycenterProblem.weights
ott.problems.linear.potentials.DualPotentials
ott.problems.linear.potentials.DualPotentials.distance
ott.problems.linear.potentials.DualPotentials.plot_ot_map
ott.problems.linear.potentials.DualPotentials.plot_potential
ott.problems.linear.potentials.DualPotentials.transport
ott.problems.linear.potentials.DualPotentials.f
ott.problems.linear.potentials.DualPotentials.g
ott.problems.linear.potentials.EntropicPotentials
ott.problems.linear.potentials.EntropicPotentials.distance
ott.problems.linear.potentials.EntropicPotentials.plot_ot_map
ott.problems.linear.potentials.EntropicPotentials.plot_potential
ott.problems.linear.potentials.EntropicPotentials.transport
ott.problems.linear.potentials.EntropicPotentials.epsilon
ott.problems.linear.potentials.EntropicPotentials.f
ott.problems.linear.potentials.EntropicPotentials.g
ott.problems.linear.potentials.EntropicPotentials.is_debiased
ott.problems.quadratic
ott.problems.quadratic.quadratic_problem.QuadraticProblem
ott.problems.quadratic.quadratic_problem.QuadraticProblem.cost_unbalanced_correction
ott.problems.quadratic.quadratic_problem.QuadraticProblem.init_transport_mass
ott.problems.quadratic.quadratic_problem.QuadraticProblem.marginal_dependent_cost
ott.problems.quadratic.quadratic_problem.QuadraticProblem.to_low_rank
ott.problems.quadratic.quadratic_problem.QuadraticProblem.update_linearization
ott.problems.quadratic.quadratic_problem.QuadraticProblem.update_lr_geom
ott.problems.quadratic.quadratic_problem.QuadraticProblem.update_lr_linearization
ott.problems.quadratic.quadratic_problem.QuadraticProblem.a
ott.problems.quadratic.quadratic_problem.QuadraticProblem.b
ott.problems.quadratic.quadratic_problem.QuadraticProblem.geom_xx
ott.problems.quadratic.quadratic_problem.QuadraticProblem.geom_xy
ott.problems.quadratic.quadratic_problem.QuadraticProblem.geom_yy
ott.problems.quadratic.quadratic_problem.QuadraticProblem.is_balanced
ott.problems.quadratic.quadratic_problem.QuadraticProblem.is_fused
ott.problems.quadratic.quadratic_problem.QuadraticProblem.is_low_rank
ott.problems.quadratic.quadratic_problem.QuadraticProblem.linear_loss
ott.problems.quadratic.quadratic_problem.QuadraticProblem.quad_loss
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.update_barycenter
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.update_features
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.flattened_b
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.flattened_y
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.gw_loss
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.is_fused
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.max_measure_size
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.ndim
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.ndim_fused
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.num_measures
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.segmented_y_b
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.segmented_y_fused
ott.problems.quadratic.gw_barycenter.GWBarycenterProblem.weights
ott.problems.quadratic.quadratic_costs.GWLoss
ott.problems.quadratic.quadratic_costs.GWLoss.count
ott.problems.quadratic.quadratic_costs.GWLoss.index
ott.problems.quadratic.quadratic_costs.GWLoss.f1
ott.problems.quadratic.quadratic_costs.GWLoss.f2
ott.problems.quadratic.quadratic_costs.GWLoss.h1
ott.problems.quadratic.quadratic_costs.GWLoss.h2
ott.problems.quadratic.quadratic_costs.make_square_loss
ott.problems.quadratic.quadratic_costs.make_kl_loss
ott.solvers
ott.solvers.linear
ott.solvers.linear.solve
ott.solvers.linear.sinkhorn.Sinkhorn
ott.solvers.linear.sinkhorn.Sinkhorn.init_state
ott.solvers.linear.sinkhorn.Sinkhorn.kernel_step
ott.solvers.linear.sinkhorn.Sinkhorn.lse_step
ott.solvers.linear.sinkhorn.Sinkhorn.one_iteration
ott.solvers.linear.sinkhorn.Sinkhorn.output_from_state
ott.solvers.linear.sinkhorn.Sinkhorn.norm_error
ott.solvers.linear.sinkhorn.Sinkhorn.outer_iterations
ott.solvers.linear.sinkhorn.SinkhornState
ott.solvers.linear.sinkhorn.SinkhornState.compute_kl_reg_cost
ott.solvers.linear.sinkhorn.SinkhornState.count
ott.solvers.linear.sinkhorn.SinkhornState.index
ott.solvers.linear.sinkhorn.SinkhornState.recenter
ott.solvers.linear.sinkhorn.SinkhornState.set
ott.solvers.linear.sinkhorn.SinkhornState.solution_error
ott.solvers.linear.sinkhorn.SinkhornState.errors
ott.solvers.linear.sinkhorn.SinkhornState.fu
ott.solvers.linear.sinkhorn.SinkhornState.gv
ott.solvers.linear.sinkhorn.SinkhornState.old_fus
ott.solvers.linear.sinkhorn.SinkhornState.old_mapped_fus
ott.solvers.linear.sinkhorn.SinkhornState.potentials
ott.solvers.linear.sinkhorn.SinkhornOutput
ott.solvers.linear.sinkhorn.SinkhornOutput.apply
ott.solvers.linear.sinkhorn.SinkhornOutput.cost_at_geom
ott.solvers.linear.sinkhorn.SinkhornOutput.count
ott.solvers.linear.sinkhorn.SinkhornOutput.index
ott.solvers.linear.sinkhorn.SinkhornOutput.marginal
ott.solvers.linear.sinkhorn.SinkhornOutput.set
ott.solvers.linear.sinkhorn.SinkhornOutput.set_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.to_dual_potentials
ott.solvers.linear.sinkhorn.SinkhornOutput.transport_cost_at_geom
ott.solvers.linear.sinkhorn.SinkhornOutput.a
ott.solvers.linear.sinkhorn.SinkhornOutput.b
ott.solvers.linear.sinkhorn.SinkhornOutput.converged
ott.solvers.linear.sinkhorn.SinkhornOutput.dual_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.ent_reg_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.errors
ott.solvers.linear.sinkhorn.SinkhornOutput.f
ott.solvers.linear.sinkhorn.SinkhornOutput.g
ott.solvers.linear.sinkhorn.SinkhornOutput.geom
ott.solvers.linear.sinkhorn.SinkhornOutput.inner_iterations
ott.solvers.linear.sinkhorn.SinkhornOutput.kl_reg_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.matrix
ott.solvers.linear.sinkhorn.SinkhornOutput.n_iters
ott.solvers.linear.sinkhorn.SinkhornOutput.ot_prob
ott.solvers.linear.sinkhorn.SinkhornOutput.potentials
ott.solvers.linear.sinkhorn.SinkhornOutput.primal_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.reg_ot_cost
ott.solvers.linear.sinkhorn.SinkhornOutput.scalings
ott.solvers.linear.sinkhorn.SinkhornOutput.threshold
ott.solvers.linear.sinkhorn.SinkhornOutput.transport_mass
ott.solvers.linear.sinkhorn_lr.LRSinkhorn
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.dykstra_update_kernel
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.dykstra_update_lse
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.init_state
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.kernel_step
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.lse_step
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.one_iteration
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.output_from_state
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.norm_error
ott.solvers.linear.sinkhorn_lr.LRSinkhorn.outer_iterations
ott.solvers.linear.sinkhorn_lr.LRSinkhornState
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.compute_error
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.count
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.index
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.reg_ot_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.set
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.solution_error
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.costs
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.crossed_threshold
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.errors
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.g
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.gamma
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.q
ott.solvers.linear.sinkhorn_lr.LRSinkhornState.r
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.apply
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.compute_reg_ot_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.cost_at_geom
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.count
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.index
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.marginal
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.set
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.set_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.transport_cost_at_geom
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.a
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.b
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.converged
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.costs
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.epsilon
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.errors
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.g
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.geom
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.inner_iterations
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.matrix
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.n_iters
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.ot_prob
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.primal_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.q
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.r
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.reg_ot_cost
ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.transport_mass
ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter
ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter.init_state
ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter.output_from_state
ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter.is_low_rank
ott.solvers.linear.continuous_barycenter.FreeWassersteinBarycenter.rank
ott.solvers.linear.continuous_barycenter.FreeBarycenterState
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.count
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.index
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.set
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.update
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.a
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.costs
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.errors
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.linear_convergence
ott.solvers.linear.continuous_barycenter.FreeBarycenterState.x
ott.solvers.linear.discrete_barycenter.FixedBarycenter
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.count
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.index
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.errors
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.f
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.g
ott.solvers.linear.discrete_barycenter.SinkhornBarycenterOutput.histogram
ott.solvers.linear.solve_univariate
ott.solvers.linear.univariate.uniform_solver
ott.solvers.linear.univariate.quantile_solver
ott.solvers.linear.univariate.north_west_solver
ott.solvers.linear.univariate.UnivariateOutput
ott.solvers.linear.univariate.UnivariateOutput.count
ott.solvers.linear.univariate.UnivariateOutput.index
ott.solvers.linear.univariate.UnivariateOutput.dual_a
ott.solvers.linear.univariate.UnivariateOutput.dual_b
ott.solvers.linear.univariate.UnivariateOutput.dual_costs
ott.solvers.linear.univariate.UnivariateOutput.mass_paired_indices
ott.solvers.linear.univariate.UnivariateOutput.mean_transport_matrix
ott.solvers.linear.univariate.UnivariateOutput.ot_costs
ott.solvers.linear.univariate.UnivariateOutput.paired_indices
ott.solvers.linear.univariate.UnivariateOutput.prob
ott.solvers.linear.univariate.UnivariateOutput.transport_matrices
ott.solvers.linear.acceleration.Momentum
ott.solvers.linear.acceleration.Momentum.lehmann
ott.solvers.linear.acceleration.Momentum.weight
ott.solvers.linear.acceleration.Momentum.error_threshold
ott.solvers.linear.acceleration.Momentum.inner_iterations
ott.solvers.linear.acceleration.Momentum.start
ott.solvers.linear.acceleration.Momentum.value
ott.solvers.linear.acceleration.AndersonAcceleration
ott.solvers.linear.acceleration.AndersonAcceleration.extrapolation
ott.solvers.linear.acceleration.AndersonAcceleration.init_maps
ott.solvers.linear.acceleration.AndersonAcceleration.update
ott.solvers.linear.acceleration.AndersonAcceleration.update_history
ott.solvers.linear.acceleration.AndersonAcceleration.memory
ott.solvers.linear.acceleration.AndersonAcceleration.refresh_every
ott.solvers.linear.acceleration.AndersonAcceleration.ridge_identity
ott.solvers.linear.implicit_differentiation.ImplicitDiff
ott.solvers.linear.implicit_differentiation.ImplicitDiff.first_order_conditions
ott.solvers.linear.implicit_differentiation.ImplicitDiff.gradient
ott.solvers.linear.implicit_differentiation.ImplicitDiff.replace
ott.solvers.linear.implicit_differentiation.ImplicitDiff.solve
ott.solvers.linear.implicit_differentiation.ImplicitDiff.precondition_fun
ott.solvers.linear.implicit_differentiation.ImplicitDiff.solver
ott.solvers.linear.implicit_differentiation.ImplicitDiff.solver_kwargs
ott.solvers.linear.implicit_differentiation.ImplicitDiff.symmetric
ott.solvers.linear.implicit_differentiation.solve_jax_cg
ott.solvers.linear.lineax_implicit.solve_lineax
ott.solvers.linear.lr_utils.unbalanced_dykstra_lse
ott.solvers.linear.lr_utils.unbalanced_dykstra_kernel
ott.solvers.quadratic
ott.solvers.quadratic.solve
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.init_state
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.output_from_state
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.is_low_rank
ott.solvers.quadratic.gromov_wasserstein.GromovWasserstein.rank
ott.solvers.quadratic.gromov_wasserstein.GWOutput
ott.solvers.quadratic.gromov_wasserstein.GWOutput.apply
ott.solvers.quadratic.gromov_wasserstein.GWOutput.count
ott.solvers.quadratic.gromov_wasserstein.GWOutput.index
ott.solvers.quadratic.gromov_wasserstein.GWOutput.set
ott.solvers.quadratic.gromov_wasserstein.GWOutput.converged
ott.solvers.quadratic.gromov_wasserstein.GWOutput.costs
ott.solvers.quadratic.gromov_wasserstein.GWOutput.errors
ott.solvers.quadratic.gromov_wasserstein.GWOutput.geom
ott.solvers.quadratic.gromov_wasserstein.GWOutput.linear_convergence
ott.solvers.quadratic.gromov_wasserstein.GWOutput.linear_state
ott.solvers.quadratic.gromov_wasserstein.GWOutput.matrix
ott.solvers.quadratic.gromov_wasserstein.GWOutput.n_iters
ott.solvers.quadratic.gromov_wasserstein.GWOutput.old_transport_mass
ott.solvers.quadratic.gromov_wasserstein.GWOutput.primal_cost
ott.solvers.quadratic.gromov_wasserstein.GWOutput.reg_gw_cost
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.dykstra_update_kernel
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.dykstra_update_lse
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.init_state
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.kernel_step
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.lse_step
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.one_iteration
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.output_from_state
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.norm_error
ott.solvers.quadratic.gromov_wasserstein_lr.LRGromovWasserstein.outer_iterations
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.apply
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.compute_reg_gw_cost
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.cost_at_geom
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.count
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.index
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.marginal
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.set
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.set_cost
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.transport_cost_at_geom
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.a
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.b
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.converged
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.costs
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.epsilon
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.errors
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.g
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.geom
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.inner_iterations
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.matrix
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.n_iters
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.ot_prob
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.primal_cost
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.q
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.r
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.reg_gw_cost
ott.solvers.quadratic.gromov_wasserstein_lr.LRGWOutput.transport_mass
ott.solvers.quadratic.lower_bound.third_lower_bound
ott.solvers.quadratic.gw_barycenter.GWBarycenterState
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.count
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.index
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.set
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.a
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.cost
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.costs
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.costs_bary
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.errors
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.gw_convergence
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.n_iters
ott.solvers.quadratic.gw_barycenter.GWBarycenterState.x
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.init_state
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.output_from_state
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.update_state
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.is_low_rank
ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.rank
ott.solvers.was_solver.WassersteinSolver
ott.solvers.was_solver.WassersteinSolver.is_low_rank
ott.solvers.was_solver.WassersteinSolver.rank
ott.solvers.utils.match_linear
ott.solvers.utils.match_quadratic
ott.solvers.utils.sample_joint
ott.solvers.utils.sample_conditional
ott.solvers.utils.uniform_sampler
ott.initializers
ott.initializers.linear
ott.initializers.linear.initializers.DefaultInitializer
ott.initializers.linear.initializers.DefaultInitializer.init_fu
ott.initializers.linear.initializers.DefaultInitializer.init_gv
ott.initializers.linear.initializers.GaussianInitializer
ott.initializers.linear.initializers.GaussianInitializer.init_fu
ott.initializers.linear.initializers.GaussianInitializer.init_gv
ott.initializers.linear.initializers.SortingInitializer
ott.initializers.linear.initializers.SortingInitializer.init_fu
ott.initializers.linear.initializers.SortingInitializer.init_gv
ott.initializers.linear.initializers.SubsampleInitializer
ott.initializers.linear.initializers.SubsampleInitializer.init_fu
ott.initializers.linear.initializers.SubsampleInitializer.init_gv
ott.initializers.linear.initializers_lr.LRInitializer
ott.initializers.linear.initializers_lr.LRInitializer.init_g
ott.initializers.linear.initializers_lr.LRInitializer.init_q
ott.initializers.linear.initializers_lr.LRInitializer.init_r
ott.initializers.linear.initializers_lr.LRInitializer.rank
ott.initializers.linear.initializers_lr.RandomInitializer
ott.initializers.linear.initializers_lr.RandomInitializer.init_g
ott.initializers.linear.initializers_lr.RandomInitializer.init_q
ott.initializers.linear.initializers_lr.RandomInitializer.init_r
ott.initializers.linear.initializers_lr.RandomInitializer.rank
ott.initializers.linear.initializers_lr.Rank2Initializer
ott.initializers.linear.initializers_lr.Rank2Initializer.init_g
ott.initializers.linear.initializers_lr.Rank2Initializer.init_q
ott.initializers.linear.initializers_lr.Rank2Initializer.init_r
ott.initializers.linear.initializers_lr.Rank2Initializer.rank
ott.initializers.linear.initializers_lr.KMeansInitializer
ott.initializers.linear.initializers_lr.KMeansInitializer.init_g
ott.initializers.linear.initializers_lr.KMeansInitializer.init_q
ott.initializers.linear.initializers_lr.KMeansInitializer.init_r
ott.initializers.linear.initializers_lr.KMeansInitializer.rank
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.init_g
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.init_q
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.init_r
ott.initializers.linear.initializers_lr.GeneralizedKMeansInitializer.rank
ott.initializers.quadratic
ott.initializers.quadratic.initializers.QuadraticInitializer
ott.neural
ott.neural.datasets
ott.neural.datasets.OTData
ott.neural.datasets.OTData.condition
ott.neural.datasets.OTData.lin
ott.neural.datasets.OTData.quad
ott.neural.datasets.OTDataset
ott.neural.datasets.OTDataset.SRC_PREFIX
ott.neural.datasets.OTDataset.TGT_PREFIX
ott.neural.methods
ott.neural.methods.monge_gap.monge_gap
ott.neural.methods.monge_gap.monge_gap_from_samples
ott.neural.methods.monge_gap.MongeGapEstimator
ott.neural.methods.monge_gap.MongeGapEstimator.setup
ott.neural.methods.monge_gap.MongeGapEstimator.train_map_estimator
ott.neural.methods.monge_gap.MongeGapEstimator.fitting_loss
ott.neural.methods.monge_gap.MongeGapEstimator.regularizer
ott.neural.methods.neuraldual.W2NeuralDual
ott.neural.methods.neuraldual.W2NeuralDual.get_step_fn
ott.neural.methods.neuraldual.W2NeuralDual.setup
ott.neural.methods.neuraldual.W2NeuralDual.to_dual_potentials
ott.neural.methods.neuraldual.W2NeuralDual.train_neuraldual_alternating
ott.neural.methods.neuraldual.W2NeuralDual.train_neuraldual_parallel
ott.neural.methods.expectile_neural_dual.ExpectileNeuralDual
ott.neural.methods.expectile_neural_dual.ExpectileNeuralDual.to_dual_potentials
ott.neural.methods.expectile_neural_dual.ExpectileNeuralDual.train_fn
ott.neural.methods.expectile_neural_dual.ENOTPotentials
ott.neural.methods.expectile_neural_dual.ENOTPotentials.distance
ott.neural.methods.expectile_neural_dual.ENOTPotentials.plot_ot_map
ott.neural.methods.expectile_neural_dual.ENOTPotentials.plot_potential
ott.neural.methods.expectile_neural_dual.ENOTPotentials.transport
ott.neural.methods.expectile_neural_dual.ENOTPotentials.f
ott.neural.methods.expectile_neural_dual.ENOTPotentials.g
ott.neural.networks
ott.neural.networks.icnn.ICNN
ott.neural.networks.icnn.ICNN.apply
ott.neural.networks.icnn.ICNN.clone
ott.neural.networks.icnn.ICNN.copy
ott.neural.networks.icnn.ICNN.create_train_state
ott.neural.networks.icnn.ICNN.get_variable
ott.neural.networks.icnn.ICNN.has_rng
ott.neural.networks.icnn.ICNN.has_variable
ott.neural.networks.icnn.ICNN.init
ott.neural.networks.icnn.ICNN.init_fn
ott.neural.networks.icnn.ICNN.init_with_output
ott.neural.networks.icnn.ICNN.is_initializing
ott.neural.networks.icnn.ICNN.is_mutable_collection
ott.neural.networks.icnn.ICNN.lazy_init
ott.neural.networks.icnn.ICNN.make_rng
ott.neural.networks.icnn.ICNN.param
ott.neural.networks.icnn.ICNN.perturb
ott.neural.networks.icnn.ICNN.potential_gradient_fn
ott.neural.networks.icnn.ICNN.potential_value_fn
ott.neural.networks.icnn.ICNN.put_variable
ott.neural.networks.icnn.ICNN.setup
ott.neural.networks.icnn.ICNN.sow
ott.neural.networks.icnn.ICNN.unbind
ott.neural.networks.icnn.ICNN.variable
ott.neural.networks.icnn.ICNN.act_fn
ott.neural.networks.icnn.ICNN.gaussian_map_samples
ott.neural.networks.icnn.ICNN.is_potential
ott.neural.networks.icnn.ICNN.name
ott.neural.networks.icnn.ICNN.parent
ott.neural.networks.icnn.ICNN.path
ott.neural.networks.icnn.ICNN.pos_weights
ott.neural.networks.icnn.ICNN.ranks
ott.neural.networks.icnn.ICNN.rectifier_fn
ott.neural.networks.icnn.ICNN.scope
ott.neural.networks.icnn.ICNN.variables
ott.neural.networks.icnn.ICNN.dim_data
ott.neural.networks.icnn.ICNN.dim_hidden
ott.neural.networks.velocity_field.VelocityField
ott.neural.networks.velocity_field.VelocityField.act_fn
ott.neural.networks.velocity_field.VelocityField.apply
ott.neural.networks.velocity_field.VelocityField.clone
ott.neural.networks.velocity_field.VelocityField.copy
ott.neural.networks.velocity_field.VelocityField.create_train_state
ott.neural.networks.velocity_field.VelocityField.get_variable
ott.neural.networks.velocity_field.VelocityField.has_rng
ott.neural.networks.velocity_field.VelocityField.has_variable
ott.neural.networks.velocity_field.VelocityField.init
ott.neural.networks.velocity_field.VelocityField.init_with_output
ott.neural.networks.velocity_field.VelocityField.is_initializing
ott.neural.networks.velocity_field.VelocityField.is_mutable_collection
ott.neural.networks.velocity_field.VelocityField.lazy_init
ott.neural.networks.velocity_field.VelocityField.make_rng
ott.neural.networks.velocity_field.VelocityField.param
ott.neural.networks.velocity_field.VelocityField.perturb
ott.neural.networks.velocity_field.VelocityField.put_variable
ott.neural.networks.velocity_field.VelocityField.setup
ott.neural.networks.velocity_field.VelocityField.sow
ott.neural.networks.velocity_field.VelocityField.time_encoder
ott.neural.networks.velocity_field.VelocityField.unbind
ott.neural.networks.velocity_field.VelocityField.variable
ott.neural.networks.velocity_field.VelocityField.condition_dims
ott.neural.networks.velocity_field.VelocityField.dropout_rate
ott.neural.networks.velocity_field.VelocityField.name
ott.neural.networks.velocity_field.VelocityField.parent
ott.neural.networks.velocity_field.VelocityField.path
ott.neural.networks.velocity_field.VelocityField.scope
ott.neural.networks.velocity_field.VelocityField.time_dims
ott.neural.networks.velocity_field.VelocityField.variables
ott.neural.networks.velocity_field.VelocityField.hidden_dims
ott.neural.networks.velocity_field.VelocityField.output_dims
ott.neural.networks.potentials.BasePotential
ott.neural.networks.potentials.BasePotential.apply
ott.neural.networks.potentials.BasePotential.clone
ott.neural.networks.potentials.BasePotential.copy
ott.neural.networks.potentials.BasePotential.create_train_state
ott.neural.networks.potentials.BasePotential.get_variable
ott.neural.networks.potentials.BasePotential.has_rng
ott.neural.networks.potentials.BasePotential.has_variable
ott.neural.networks.potentials.BasePotential.init
ott.neural.networks.potentials.BasePotential.init_with_output
ott.neural.networks.potentials.BasePotential.is_initializing
ott.neural.networks.potentials.BasePotential.is_mutable_collection
ott.neural.networks.potentials.BasePotential.lazy_init
ott.neural.networks.potentials.BasePotential.make_rng
ott.neural.networks.potentials.BasePotential.param
ott.neural.networks.potentials.BasePotential.perturb
ott.neural.networks.potentials.BasePotential.potential_gradient_fn
ott.neural.networks.potentials.BasePotential.potential_value_fn
ott.neural.networks.potentials.BasePotential.put_variable
ott.neural.networks.potentials.BasePotential.setup
ott.neural.networks.potentials.BasePotential.sow
ott.neural.networks.potentials.BasePotential.unbind
ott.neural.networks.potentials.BasePotential.variable
ott.neural.networks.potentials.BasePotential.is_potential
ott.neural.networks.potentials.BasePotential.name
ott.neural.networks.potentials.BasePotential.parent
ott.neural.networks.potentials.BasePotential.path
ott.neural.networks.potentials.BasePotential.scope
ott.neural.networks.potentials.BasePotential.variables
ott.neural.networks.potentials.PotentialMLP
ott.neural.networks.potentials.PotentialMLP.act_fn
ott.neural.networks.potentials.PotentialMLP.apply
ott.neural.networks.potentials.PotentialMLP.clone
ott.neural.networks.potentials.PotentialMLP.copy
ott.neural.networks.potentials.PotentialMLP.create_train_state
ott.neural.networks.potentials.PotentialMLP.get_variable
ott.neural.networks.potentials.PotentialMLP.has_rng
ott.neural.networks.potentials.PotentialMLP.has_variable
ott.neural.networks.potentials.PotentialMLP.init
ott.neural.networks.potentials.PotentialMLP.init_with_output
ott.neural.networks.potentials.PotentialMLP.is_initializing
ott.neural.networks.potentials.PotentialMLP.is_mutable_collection
ott.neural.networks.potentials.PotentialMLP.lazy_init
ott.neural.networks.potentials.PotentialMLP.make_rng
ott.neural.networks.potentials.PotentialMLP.param
ott.neural.networks.potentials.PotentialMLP.perturb
ott.neural.networks.potentials.PotentialMLP.potential_gradient_fn
ott.neural.networks.potentials.PotentialMLP.potential_value_fn
ott.neural.networks.potentials.PotentialMLP.put_variable
ott.neural.networks.potentials.PotentialMLP.setup
ott.neural.networks.potentials.PotentialMLP.sow
ott.neural.networks.potentials.PotentialMLP.unbind
ott.neural.networks.potentials.PotentialMLP.variable
ott.neural.networks.potentials.PotentialMLP.is_potential
ott.neural.networks.potentials.PotentialMLP.name
ott.neural.networks.potentials.PotentialMLP.parent
ott.neural.networks.potentials.PotentialMLP.path
ott.neural.networks.potentials.PotentialMLP.scope
ott.neural.networks.potentials.PotentialMLP.variables
ott.neural.networks.potentials.PotentialMLP.dim_hidden
ott.neural.networks.potentials.MLP
ott.neural.networks.potentials.MLP.act_fn
ott.neural.networks.potentials.MLP.apply
ott.neural.networks.potentials.MLP.clone
ott.neural.networks.potentials.MLP.copy
ott.neural.networks.potentials.MLP.get_variable
ott.neural.networks.potentials.MLP.has_rng
ott.neural.networks.potentials.MLP.has_variable
ott.neural.networks.potentials.MLP.init
ott.neural.networks.potentials.MLP.init_with_output
ott.neural.networks.potentials.MLP.is_initializing
ott.neural.networks.potentials.MLP.is_mutable_collection
ott.neural.networks.potentials.MLP.lazy_init
ott.neural.networks.potentials.MLP.make_rng
ott.neural.networks.potentials.MLP.param
ott.neural.networks.potentials.MLP.perturb
ott.neural.networks.potentials.MLP.put_variable
ott.neural.networks.potentials.MLP.setup
ott.neural.networks.potentials.MLP.sow
ott.neural.networks.potentials.MLP.unbind
ott.neural.networks.potentials.MLP.variable
ott.neural.networks.potentials.MLP.name
ott.neural.networks.potentials.MLP.parent
ott.neural.networks.potentials.MLP.path
ott.neural.networks.potentials.MLP.scope
ott.neural.networks.potentials.MLP.variables
ott.neural.networks.potentials.MLP.dim_hidden
ott.neural.networks.potentials.PotentialTrainState
ott.neural.networks.potentials.PotentialTrainState.apply_gradients
ott.neural.networks.potentials.PotentialTrainState.create
ott.neural.networks.potentials.PotentialTrainState.replace
ott.neural.networks.potentials.PotentialTrainState.potential_value_fn
ott.neural.networks.potentials.PotentialTrainState.potential_gradient_fn
ott.neural.networks.potentials.PotentialTrainState.step
ott.neural.networks.potentials.PotentialTrainState.apply_fn
ott.neural.networks.potentials.PotentialTrainState.params
ott.neural.networks.potentials.PotentialTrainState.tx
ott.neural.networks.potentials.PotentialTrainState.opt_state
ott.experimental
ott.experimental.mmsinkhorn
ott.experimental.mmsinkhorn.MMSinkhorn
ott.experimental.mmsinkhorn.MMSinkhorn.init_state
ott.experimental.mmsinkhorn.MMSinkhorn.outer_iterations
ott.experimental.mmsinkhorn.MMSinkhornOutput
ott.experimental.mmsinkhorn.MMSinkhornOutput.count
ott.experimental.mmsinkhorn.MMSinkhornOutput.index
ott.experimental.mmsinkhorn.MMSinkhornOutput.marginal
ott.experimental.mmsinkhorn.MMSinkhornOutput.set
ott.experimental.mmsinkhorn.MMSinkhornOutput.a_s
ott.experimental.mmsinkhorn.MMSinkhornOutput.converged
ott.experimental.mmsinkhorn.MMSinkhornOutput.cost_fns
ott.experimental.mmsinkhorn.MMSinkhornOutput.cost_t
ott.experimental.mmsinkhorn.MMSinkhornOutput.ent_reg_cost
ott.experimental.mmsinkhorn.MMSinkhornOutput.epsilon
ott.experimental.mmsinkhorn.MMSinkhornOutput.errors
ott.experimental.mmsinkhorn.MMSinkhornOutput.inner_iterations
ott.experimental.mmsinkhorn.MMSinkhornOutput.marginals
ott.experimental.mmsinkhorn.MMSinkhornOutput.n_iters
ott.experimental.mmsinkhorn.MMSinkhornOutput.n_marginals
ott.experimental.mmsinkhorn.MMSinkhornOutput.potentials
ott.experimental.mmsinkhorn.MMSinkhornOutput.shape
ott.experimental.mmsinkhorn.MMSinkhornOutput.tensor
ott.experimental.mmsinkhorn.MMSinkhornOutput.threshold
ott.experimental.mmsinkhorn.MMSinkhornOutput.transport_mass
ott.experimental.mmsinkhorn.MMSinkhornOutput.x_s
ott.tools
ott.tools.unreg.hungarian
ott.tools.unreg.HungarianOutput
ott.tools.unreg.HungarianOutput.count
ott.tools.unreg.HungarianOutput.index
ott.tools.unreg.HungarianOutput.geom
ott.tools.unreg.HungarianOutput.matrix
ott.tools.unreg.HungarianOutput.paired_indices
ott.tools.unreg.wassdis_p
ott.tools.segment_sinkhorn.segment_sinkhorn
ott.tools.sinkhorn_divergence.sinkdiv
ott.tools.sinkhorn_divergence.sinkhorn_divergence
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.to_dual_potentials
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.is_low_rank
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.divergence
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.geoms
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.a
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.b
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.potentials
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.factors
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.errors
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.converged
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.n_iters
ott.tools.sinkhorn_divergence.segment_sinkhorn_divergence
ott.tools.sliced.random_proj_sphere
ott.tools.sliced.sliced_wasserstein
ott.tools.progot.ProgOT
ott.tools.progot.ProgOTOutput
ott.tools.progot.ProgOTOutput.count
ott.tools.progot.ProgOTOutput.get_output
ott.tools.progot.ProgOTOutput.index
ott.tools.progot.ProgOTOutput.transport
ott.tools.progot.ProgOTOutput.alphas
ott.tools.progot.ProgOTOutput.converged
ott.tools.progot.ProgOTOutput.epsilons
ott.tools.progot.ProgOTOutput.is_debiased
ott.tools.progot.ProgOTOutput.num_iters
ott.tools.progot.ProgOTOutput.num_steps
ott.tools.progot.ProgOTOutput.outputs
ott.tools.progot.ProgOTOutput.prob
ott.tools.progot.ProgOTOutput.xs
ott.tools.progot.get_alpha_schedule
ott.tools.progot.get_epsilon_schedule
ott.tools.soft_sort.multivariate_cdf_quantile_maps
ott.tools.soft_sort.quantile
ott.tools.soft_sort.quantile_normalization
ott.tools.soft_sort.quantize
ott.tools.soft_sort.ranks
ott.tools.soft_sort.sort
ott.tools.soft_sort.sort_with
ott.tools.soft_sort.topk_mask
ott.tools.k_means.k_means
ott.tools.k_means.KMeansOutput
ott.tools.k_means.KMeansOutput.count
ott.tools.k_means.KMeansOutput.index
ott.tools.k_means.KMeansOutput.assignment
ott.tools.k_means.KMeansOutput.centroids
ott.tools.k_means.KMeansOutput.converged
ott.tools.k_means.KMeansOutput.error
ott.tools.k_means.KMeansOutput.inner_errors
ott.tools.k_means.KMeansOutput.iteration
ott.tools.plot.Plot
ott.tools.plot.Plot.animate
ott.tools.plot.Plot.update
ott.tools.gaussian_mixture.gaussian.Gaussian
ott.tools.gaussian_mixture.gaussian.Gaussian.covariance
ott.tools.gaussian_mixture.gaussian.Gaussian.f_potential
ott.tools.gaussian_mixture.gaussian.Gaussian.from_mean_and_cov
ott.tools.gaussian_mixture.gaussian.Gaussian.from_random
ott.tools.gaussian_mixture.gaussian.Gaussian.from_samples
ott.tools.gaussian_mixture.gaussian.Gaussian.from_z
ott.tools.gaussian_mixture.gaussian.Gaussian.log_prob
ott.tools.gaussian_mixture.gaussian.Gaussian.sample
ott.tools.gaussian_mixture.gaussian.Gaussian.to_z
ott.tools.gaussian_mixture.gaussian.Gaussian.transport
ott.tools.gaussian_mixture.gaussian.Gaussian.w2_dist
ott.tools.gaussian_mixture.gaussian.Gaussian.loc
ott.tools.gaussian_mixture.gaussian.Gaussian.n_dimensions
ott.tools.gaussian_mixture.gaussian.Gaussian.scale
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.components
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.conditional_log_prob
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.from_mean_cov_component_weights
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.from_points_and_assignment_probs
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.from_random
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.get_component
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.get_log_component_posterior
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.has_nans
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.log_component_weights
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.log_prob
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.sample
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.cholesky
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.component_weight_ob
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.component_weights
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.covariance
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.dtype
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.loc
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.n_components
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.n_dimensions
ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.scale_params
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_bures_geometry
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_cost_matrix
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_normalized_sinkhorn_coupling
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_sinkhorn
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.dtype
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.epsilon
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.gmm0
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.gmm1
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.lock_gmm1
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.rho
ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.tau
ott.tools.gaussian_mixture.fit_gmm.initialize
ott.tools.gaussian_mixture.fit_gmm.fit_model_em
ott.tools.gaussian_mixture.fit_gmm_pair.get_fit_model_em_fn
ott.math
ott.math.fixed_point_loop.fixpoint_iter
ott.math.matrix_square_root.sqrtm
ott.math.utils.norm
ott.math.utils.logsumexp
ott.math.utils.softmin
ott.math.utils.lambertw
ott.utils
ott.utils.default_progress_fn
ott.utils.tqdm_progress_fn
ott.utils.batched_vmap
References
Glossary
Bibliography
Contributing Guide
.rst
.pdf
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.a
Contents
SinkhornDivergenceOutput.a
ott.tools.sinkhorn_divergence.SinkhornDivergenceOutput.a
#
SinkhornDivergenceOutput.
a
:
Array
#
Contents
SinkhornDivergenceOutput.a
