{"categories":[{"label":"Mathematics","url":"https://skillfed.io/packages/category/scientific-engineering-mathematics/3"}],"enrichment":{"capability":"OTT-JAX solves optimal transport problems\u2014matching and comparing point clouds\u2014using JAX-accelerated algorithms like Sinkhorn, including support for Gromov-Wasserstein and Wasserstein barycenter computations.","skillfed_tags":["optimal-transport","jax-accelerated","point-cloud-matching"],"use_cases":["Compute the optimal coupling matrix between two point clouds to measure similarity or perform alignment.","Compare distributions in different geometric spaces using Gromov-Wasserstein distance for domain adaptation or shape matching.","Train a neural network to approximate the optimal transport map between two measures for fast inference on new data.","Compute Wasserstein barycenters to find representative point sets that minimize transport cost to a collection of input distributions.","Integrate optimal transport into machine learning pipelines via JAX's autodiff for end-to-end gradient-based optimization."],"what_it_does":"OTT-JAX is a JAX-powered library for solving optimal transport problems at scale on accelerators. It implements discrete solvers\u2014notably the Sinkhorn algorithm with various optimizations (scheduling, momentum, low-rank approximations)\u2014to compute couplings between point clouds. Beyond basic matching, it supports advanced problems like Gromov-Wasserstein distance (comparing point clouds in different geometric spaces) and Wasserstein barycenters (finding representative point sets). The library also includes neural network approaches that learn transport maps between measure pairs, enabling both exact and approximate solutions depending on problem size and accuracy needs.\n\nThe package is built on six core dependencies: jax for autodifferentiation and GPU/TPU acceleration, jaxopt and optax for optimization, lineax for linear algebra, numpy for array operations, and typing_extensions for type hints. It supports Python 3.9 through 3.13 and runs on Linux, macOS, and Windows. The codebase is maintained by researchers at Apple with contributions from Google, Meta, and academic partners, reflecting active development and research backing.","worth_installing":"Yes. OTT-JAX is actively maintained, has low install friction, carries a permissive Apache 2.0 license, and implements a mature, well-documented suite of optimal transport algorithms. It is suitable for research, production machine learning, and scientific computing where point cloud matching, distribution comparison, or transport-based optimization is needed. No known security vulnerabilities."},"id":"ott-jax","links":{"html":"https://skillfed.io/packages/ott-jax","md":"https://skillfed.io/packages/ott-jax.md","pypi":"https://pypi.org/project/ott-jax/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2025-11-04","license_spdx":null,"license_treatment":"permissive","name":"ott-jax","python_support":"supports_current","summary":"Optimal Transport Tools in JAX"},"popularity":{"monthly_downloads":94080,"position":13344,"tier":"top_15000"},"security":{"n_vulnerabilities":0},"version":"0.6.0"}