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}
翻译:最佳运输工具(OTT-JAX)是一个能解决点云和直方图之间最佳运输问题的Python工具箱,该工具箱以各种JAX特性为基础,如自动和定制反向模式差异、矢量化、即时编集和加速器支持。工具箱包括基本计算,如正规化的OT问题的解决,以及更先进的扩展,如嘉宾中心、Gromov-Wasserstein、低级解答器、对convex地图的估计、可区别的量化和等级的概括化以及高斯混合物之间的近似OT。该工具箱代码可在\ textt{https://github.com/ott-jax/ott}查阅。