ilyaraz / ot_estimators
Implementations of several fast approximate algorithms for geometric optimal transport (OT)
☆117Updated 4 years ago
Alternatives and similar repositories for ot_estimators:
Users that are interested in ot_estimators are comparing it to the libraries listed below
- fast kernel evaluation in high dimensions via hashing☆23Updated 4 years ago
- [NeurIPS 2019 Spotlight] High dimensional mean estimation and outlier detection in nearly-linear time.☆26Updated 5 years ago
- Experiment code for "Randomized Automatic Differentiation"☆66Updated 4 years ago
- Python implementation of smooth optimal transport.☆57Updated 3 years ago
- Official implementation (learning part) for paper: Numerically Accurate Hyperbolic Embeddings Using Tiling-Based Models☆18Updated 6 months ago
- Efficient LSH-based kernel density estimation☆29Updated 5 years ago
- A collection of adaptive sparse multi-scale solvers for optimal transport and related optimization problems.☆53Updated 3 years ago
- A library for exploring persistent homology☆100Updated 2 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆104Updated 4 years ago
- Persistent Homology Algorithm Toolbox (PHAT).☆35Updated 5 years ago
- Neural LSH [ICLR 2020] - Using supervised learning to produce better space partitions for fast nearest neighbor search.☆73Updated 4 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- Fast optimal transport code☆55Updated 4 years ago
- FALKON implementation used in the experimental section of "FALKON: An Optimal Large Scale Kernel Method"☆29Updated 4 years ago
- Hessian spectral density estimation in TF and Jax☆121Updated 4 years ago
- Optimization with orthogonal constraints and on general manifolds☆127Updated 4 years ago
- Python implementation of projection losses.☆25Updated 5 years ago
- MMD, Hausdorff and Sinkhorn divergences scaled up to 1,000,000 samples.☆55Updated 5 years ago
- ☆98Updated 3 years ago
- A manifold optimization library for deep learning☆241Updated 3 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- Code for "Efficient optimization of loops and limits with randomized telescoping sums"☆27Updated 5 years ago
- Functional TensorFlow Implementation of Singular Value Decomposition for paper Fast Graph Learning☆13Updated 3 years ago
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆115Updated last year
- Code for fast dpsgd implementations in JAX/TF☆59Updated 2 years ago
- ☆213Updated 3 years ago
- Pytorch implementation of the Power Spherical distribution☆74Updated 7 months ago
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆64Updated 4 months ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 4 years ago