论文标题
最佳运输工具（OTT）：所有事物的JAX工具箱
Optimal Transport Tools (OTT): A JAX Toolbox for all things Wasserstein
论文作者
论文摘要
最佳传输工具（OTT-JAX）是一个Python工具箱，可以解决点云和直方图之间的最佳传输问题。该工具箱基于各种JAX功能，例如自动和自定义反向模式差异化，矢量化，即时编译和加速器支持。该工具箱涵盖了基本计算，例如正则化ot问题的分辨率以及更高级的扩展，例如barycenters，gromov-wasserstein，低级别求解器，凸图的估计，对位点和等级的可区分概括，以及高斯混合物之间的近似OT。该工具箱代码可在\ texttt {https://github.com/ott-jax/ott}获得
Optimal transport tools (OTT-JAX) is a Python toolbox that can solve optimal transport problems between point clouds and histograms. The toolbox builds on various JAX features, such as automatic and custom reverse mode differentiation, vectorization, just-in-time compilation and accelerators support. The toolbox covers elementary computations, such as the resolution of the regularized OT problem, and more advanced extensions, such as barycenters, Gromov-Wasserstein, low-rank solvers, estimation of convex maps, differentiable generalizations of quantiles and ranks, and approximate OT between Gaussian mixtures. The toolbox code is available at \texttt{https://github.com/ott-jax/ott}
