gpeyre / ot4mlLinks
Optimal Transport for Machine Learners
☆43Updated 2 months ago
Alternatives and similar repositories for ot4ml
Users that are interested in ot4ml are comparing it to the libraries listed below
Sorting:
- Neural Optimal Transport with Lagrangian Costs☆61Updated 7 months ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆36Updated 9 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆152Updated 2 months ago
- MoMo: Momentum Models for Adaptive Learning Rates☆19Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆108Updated last year
- Algorithms for computations on random manifolds made easier☆94Updated 2 years ago
- ☆36Updated 9 months ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- Comparison between GFlowNets & Maximum Entropy RL☆19Updated last year
- ☆23Updated 2 years ago
- code for "Riemannian Flow Matching on General Geometries".☆282Updated last year
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆41Updated last year
- ☆14Updated 2 years ago
- Tutorial on amortized optimization for learning to optimize over continuous domains☆249Updated 3 months ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆44Updated 2 years ago
- Mutual information estimators and benchmark☆56Updated 3 months ago
- Neural Diffusion Processes☆81Updated last year
- Repository for "Generative Flow Networks as Entropy-Regularized RL" (AISTATS-2024, Oral)☆40Updated last year
- ☆46Updated 2 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆40Updated 3 years ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- Code for the paper "Beyond Deep Ensembles: A Large-Scale Evaluation of Bayesian Deep Learning under Distribution Shift"☆22Updated 2 years ago
- A minimal example of optimal transport with Input Convex Neural Networks in Pytorch☆24Updated 4 years ago
- This repository contains the code for the paper "Geometry-aware Bayesian Optimization in Roboticsusing Riemannian Matérn Kernels" (CoRL'2…☆21Updated 2 years ago
- Supplementary code for the NeurIPS 2020 paper "Matern Gaussian processes on Riemannian manifolds".☆29Updated 11 months ago
- Repository containing the code for Implicit Manifold Gaussian Process.☆16Updated last year
- Geometric kernels on manifolds, meshes and graphs☆283Updated last month
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 5 months ago