☆213Jan 31, 2022Updated 4 years ago
Alternatives and similar repositories for ott
Users that are interested in ott are comparing it to the libraries listed below
Sorting:
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆701Feb 5, 2026Updated 3 weeks ago
- Convex potential flows☆85Nov 19, 2021Updated 4 years ago
- Riemannian Convex Potential Maps☆67Mar 31, 2023Updated 2 years ago
- Meta Optimal Transport☆107May 25, 2023Updated 2 years ago
- The Differentiable Cross-Entropy Method☆124Aug 14, 2020Updated 5 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆47Jun 17, 2022Updated 3 years ago
- POT : Python Optimal Transport☆2,757Updated this week
- Euclidean Wasserstein-2 optimal transportation☆46Aug 19, 2023Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆42Oct 26, 2020Updated 5 years ago
- ☆74Feb 17, 2023Updated 3 years ago
- ☆620Jan 23, 2026Updated last month
- [ICLR 2022] Path integral sampler☆52Aug 21, 2023Updated 2 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆24Sep 9, 2024Updated last year
- ☆774Jan 27, 2024Updated 2 years ago
- ☆51Jul 25, 2024Updated last year
- An implementation of "Subspace Representations for Soft Set Operations and Sentence Similarities" (NAACL 2024)☆10May 31, 2024Updated last year
- Geometric loss functions between point clouds, images and volumes☆683Apr 14, 2025Updated 10 months ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- A library for programmatically generating equivariant layers through constraint solving☆281May 8, 2023Updated 2 years ago
- PyTorch implementation of the OT-Flow approach in arXiv:2006.00104☆57Jul 25, 2024Updated last year
- ☆164Dec 13, 2023Updated 2 years ago
- ☆50Oct 22, 2020Updated 5 years ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆78Mar 4, 2024Updated last year
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- ☆925Jan 29, 2026Updated last month
- Fast Differentiable Sorting and Ranking☆616Feb 15, 2024Updated 2 years ago
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,551May 2, 2024Updated last year
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated 11 months ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Sep 28, 2022Updated 3 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago
- Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…☆348Dec 21, 2021Updated 4 years ago
- A Graph Neural Network Library in Jax☆1,465Mar 18, 2024Updated last year
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆105Nov 20, 2021Updated 4 years ago
- Implementation of the Gromov-Wasserstein distance to the setting of Unbalanced Optimal Transport☆45Mar 13, 2023Updated 2 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,375Mar 1, 2024Updated 2 years ago
- Regularized Neural ODEs (RNODE)☆81Jul 16, 2021Updated 4 years ago
- ☆109Jun 25, 2024Updated last year
- Optax is a gradient processing and optimization library for JAX.☆2,193Updated this week
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago