Skip to main content
Ctrl+K
ott 0.6.1.dev25+ge241ec3d4 documentation - Home ott 0.6.1.dev25+ge241ec3d4 documentation - Home

Examples

  • Getting Started
  • Tutorials
    • Geometry
      • Point Clouds
      • Grid Geometry
    • Linear OT
      • Focus on Sinkhorn
      • OTT & POT
      • Gradient flows
      • Low-rank Sinkhorn
      • Sinkhorn Divergence Hessians
      • MBO Sparse Maps
      • Multimarginal OT
      • Progressive Entropic Optimal Transport
      • Unbalanced Optimal Transport
      • Sharded Sinkhorn
      • Semidiscrete OT
    • Quadratic OT
      • Gromov-Wasserstein
      • Low-rank GW
      • GW for Multi-omics
    • Neural OT
      • Neural Dual Solver
      • Monge gap
      • Expectile-Regularized OT
      • Meta OT
      • Flow Matching
      • Choice of couplings in flow matching
    • Barycenter
      • Sinkhorn Barycenters
      • Wasserstein Barycenter
      • GMM Barycenters
      • Fitting Coupled GMMs
    • Theory
      • Benamou-Brenier
    • Miscellaneous
      • Tracking progress of ott.solvers
      • Soft Sorting
      • Single-cell Genomics
      • Optimal Transport Conformal Prediction

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.diag_cost
      • 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_square
      • 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.diag_cost
      • 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_neg_dotp
      • ott.geometry.pointcloud.PointCloud.is_online
      • ott.geometry.pointcloud.PointCloud.is_square
      • 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.diag_cost
      • 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_square
      • 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.diag_cost
      • 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_square
      • 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.diag_cost
      • 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_square
      • 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.diag_cost
      • 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_square
      • 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.diag_cost
      • 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_square
      • 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.semidiscrete_pointcloud.SemidiscretePointCloud
      • ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud.sample
      • ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud.dtype
      • ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud.epsilon
      • ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud.is_entropy_regularized
      • ott.geometry.semidiscrete_pointcloud.SemidiscretePointCloud.shape
    • 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.NegDotProduct
      • ott.geometry.costs.NegDotProduct.all_pairs
      • ott.geometry.costs.NegDotProduct.barycenter
      • ott.geometry.costs.NegDotProduct.norm
      • ott.geometry.costs.NegDotProduct.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.potential_fn_from_dual_vec
        • 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_assignment
        • 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.semidiscrete_linear_problem.SemidiscreteLinearProblem
        • ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem.potential_fn_from_dual_vec
        • ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem.sample
        • ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem.b
        • ott.problems.linear.semidiscrete_linear_problem.SemidiscreteLinearProblem.epsilon
      • 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.num_per_measure
        • 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.DualPotentials.cost_fn
    • 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.num_per_measure
        • 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.solve_semidiscrete
      • 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.diag
        • ott.solvers.linear.sinkhorn.SinkhornOutput.dual_cost
        • ott.solvers.linear.sinkhorn.SinkhornOutput.ent_reg_cost
        • ott.solvers.linear.sinkhorn.SinkhornOutput.entropy
        • 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.normalized_entropy
        • 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.iterations
        • 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.linear_outputs
        • 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.semidiscrete.SemidiscreteSolver
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.epsilon_scheduler
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.step
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.callback
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.error_batch_size
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.error_eval_every
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.error_num_repeats
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.potential_ema
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.threshold
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.num_iterations
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.batch_size
        • ott.solvers.linear.semidiscrete.SemidiscreteSolver.optimizer
      • ott.solvers.linear.semidiscrete.SemidiscreteState
        • ott.solvers.linear.semidiscrete.SemidiscreteState.it
        • ott.solvers.linear.semidiscrete.SemidiscreteState.epsilon
        • ott.solvers.linear.semidiscrete.SemidiscreteState.g
        • ott.solvers.linear.semidiscrete.SemidiscreteState.g_ema
        • ott.solvers.linear.semidiscrete.SemidiscreteState.opt_state
        • ott.solvers.linear.semidiscrete.SemidiscreteState.losses
        • ott.solvers.linear.semidiscrete.SemidiscreteState.grad_norms
        • ott.solvers.linear.semidiscrete.SemidiscreteState.errors
      • ott.solvers.linear.semidiscrete.SemidiscreteOutput
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.marginal_chi2_error
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.sample
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.to_dataloader
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.to_dual_potentials
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.converged
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.errors
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.geom
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.it
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.losses
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.g
        • ott.solvers.linear.semidiscrete.SemidiscreteOutput.prob
      • ott.solvers.linear.semidiscrete.HardAssignmentOutput
        • ott.solvers.linear.semidiscrete.HardAssignmentOutput.dual_cost
        • ott.solvers.linear.semidiscrete.HardAssignmentOutput.f
        • ott.solvers.linear.semidiscrete.HardAssignmentOutput.g
        • ott.solvers.linear.semidiscrete.HardAssignmentOutput.geom
        • ott.solvers.linear.semidiscrete.HardAssignmentOutput.matrix
        • ott.solvers.linear.semidiscrete.HardAssignmentOutput.primal_cost
        • ott.solvers.linear.semidiscrete.HardAssignmentOutput.ot_prob
        • ott.solvers.linear.semidiscrete.HardAssignmentOutput.paired_indices
      • ott.solvers.linear.semidiscrete.constant_epsilon_scheduler
      • 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.data
      • ott.neural.data.ot_dataloader.LinearOTDataloader
        • ott.neural.data.ot_dataloader.LinearOTDataloader.cost_fn
        • ott.neural.data.ot_dataloader.LinearOTDataloader.epsilon
        • ott.neural.data.ot_dataloader.LinearOTDataloader.max_iterations
        • ott.neural.data.ot_dataloader.LinearOTDataloader.relative_epsilon
        • ott.neural.data.ot_dataloader.LinearOTDataloader.replace
        • ott.neural.data.ot_dataloader.LinearOTDataloader.shardings
        • ott.neural.data.ot_dataloader.LinearOTDataloader.threshold
        • ott.neural.data.ot_dataloader.LinearOTDataloader.rng
        • ott.neural.data.ot_dataloader.LinearOTDataloader.dataset
      • ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader
        • ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader.epsilon
        • ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader.out_shardings
        • ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader.return_indices
        • ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader.subset_size
        • ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader.subset_size_threshold
        • ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader.rng
        • ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader.sd_out
        • ott.neural.data.semidiscrete_dataloader.SemidiscreteDataloader.batch_size
    • ott.neural.methods
      • ott.neural.methods.flow_matching.flow_matching_step
      • ott.neural.methods.flow_matching.interpolate_samples
      • ott.neural.methods.flow_matching.evaluate_velocity_field
      • ott.neural.methods.flow_matching.curvature
      • ott.neural.methods.flow_matching.gaussian_nll
      • 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.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.methods.expectile_neural_dual.ENOTPotentials.cost_fn
    • ott.neural.networks
      • ott.neural.networks.icnn.ICNN
        • ott.neural.networks.icnn.ICNN.eval
        • ott.neural.networks.icnn.ICNN.gradient
        • ott.neural.networks.icnn.ICNN.iter_children
        • ott.neural.networks.icnn.ICNN.iter_modules
        • ott.neural.networks.icnn.ICNN.perturb
        • ott.neural.networks.icnn.ICNN.sow
        • ott.neural.networks.icnn.ICNN.train
        • ott.neural.networks.icnn.ICNN.is_potential
      • ott.neural.networks.icnn.KeyNet
        • ott.neural.networks.icnn.KeyNet.eval
        • ott.neural.networks.icnn.KeyNet.gradient
        • ott.neural.networks.icnn.KeyNet.iter_children
        • ott.neural.networks.icnn.KeyNet.iter_modules
        • ott.neural.networks.icnn.KeyNet.perturb
        • ott.neural.networks.icnn.KeyNet.sow
        • ott.neural.networks.icnn.KeyNet.train
        • ott.neural.networks.icnn.KeyNet.is_potential
      • 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.eval
        • ott.neural.networks.potentials.PotentialMLP.iter_children
        • ott.neural.networks.potentials.PotentialMLP.iter_modules
        • 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.sow
        • ott.neural.networks.potentials.PotentialMLP.train
        • ott.neural.networks.potentials.PotentialMLP.is_potential
      • ott.neural.networks.potentials.MLP
        • ott.neural.networks.potentials.MLP.eval
        • ott.neural.networks.potentials.MLP.iter_children
        • ott.neural.networks.potentials.MLP.iter_modules
        • ott.neural.networks.potentials.MLP.perturb
        • ott.neural.networks.potentials.MLP.sow
        • ott.neural.networks.potentials.MLP.train
      • 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.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.conformal.OTCP
      • ott.tools.conformal.OTCP.calibrate
      • ott.tools.conformal.OTCP.fit_transport
      • ott.tools.conformal.OTCP.get_scores
      • ott.tools.conformal.OTCP.nonconformity_fn
      • ott.tools.conformal.OTCP.predict
      • ott.tools.conformal.OTCP.calibration_scores
      • ott.tools.conformal.OTCP.offset
      • ott.tools.conformal.OTCP.sampler
      • ott.tools.conformal.OTCP.scale
      • ott.tools.conformal.OTCP.sinkhorn_output
      • ott.tools.conformal.OTCP.target_measure
      • ott.tools.conformal.OTCP.model
    • ott.tools.conformal.sobol_ball_sampler
    • 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.plot.PlotMM
      • ott.tools.plot.PlotMM.animate
      • ott.tools.plot.PlotMM.update
    • ott.tools.plot.transport_animation
    • 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.lbfgs
    • ott.math.legendre
    • ott.math.velocity_from_brenier_potential
    • 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

Tutorials

Tutorials#

  • Geometry
    • Point Clouds
    • Grid Geometry
  • Linear OT
    • Focus on Sinkhorn
    • OTT & POT
    • Gradient flows
    • Low-rank Sinkhorn
    • Sinkhorn Divergence Hessians
    • MBO Sparse Maps
    • Multimarginal OT
    • Progressive Entropic Optimal Transport
    • Unbalanced Optimal Transport
    • Sharded Sinkhorn
    • Semidiscrete OT
  • Quadratic OT
    • Gromov-Wasserstein
    • Low-rank GW
    • GW for Multi-omics
  • Neural OT
    • Neural Dual Solver
    • Monge gap
    • Expectile-Regularized OT
    • Meta OT
    • Flow Matching
    • Choice of couplings in flow matching
  • Barycenter
    • Sinkhorn Barycenters
    • Wasserstein Barycenter
    • GMM Barycenters
    • Fitting Coupled GMMs
  • Theory
    • Benamou-Brenier
  • Miscellaneous
    • Tracking progress of ott.solvers
    • Soft Sorting
    • Single-cell Genomics
    • Optimal Transport Conformal Prediction

previous

Getting Started

next

Geometry

By OTT authors

© Copyright 2021-2026, OTT authors.