ott
Optimal matcher
A JAX-based library for computing optimal transport problems in various settings
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
525 stars
10 watching
81 forks
Language: Python
last commit: 3 days ago
Linked from 1 awesome list
automatic-differentiationgromov-wassersteinjaxoptimal-transportsinkhorn
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 eniac/otti | A compiler and solver for optimization problems using zkSNARKs | 6 |
 google/jaxopt | An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning. | 933 |
 dipolar-quantum-gases/jaxfit | A package for GPU/TPU accelerated nonlinear least-squares curve fitting using JAX | 51 |
 stevengj/nlopt | A unified interface and packaging of several nonlinear optimization libraries for global and local optimization. | 1,892 |
 jmrichardson/tuneta | Automates optimization of technical indicators for machine learning models in finance | 413 |
 locuslab/optnet | A PyTorch module that adds differentiable optimization as a layer to neural networks | 513 |
 sidak/otfusion | Model fusion via optimal transport to combine performance of multiple machine learning models | 135 |
 hyperopt/hyperopt-sklearn | Automates search for optimal parameters in machine learning algorithms. | 1,588 |
 osqp/osqpth | A PyTorch module that wraps the OSQP solver for differentiable optimization problems | 59 |
 stunkymonkey/osm_ch | A tool that uses graph contraction hierarchies to speed up shortest path calculations on OpenStreetMap data | 17 |
 matthewpeterkelly/optimtraj | A software package that optimizes trajectories for physical systems by finding the optimal sequence of controls to minimize energy or cost | 625 |
 jiangoforit/yellowfin_pytorch | An optimizer that automatically tunes momentum and learning rate using local quadratic approximation. | 287 |
 jycouet/kitql | A collection of standalone tools to speedrun web applications built with GraphQL and Svelte | 407 |
 metaopt/torchopt | An efficient library for differentiable optimization built on top of PyTorch. | 544 |
 stonesjtu/pytorch-nce | An implementation of Noise Contrastive Estimation in PyTorch to improve the performance and efficiency of softmax output layers | 318 |