ilyaraz / ot_estimators
Implementations of several fast approximate algorithms for geometric optimal transport (OT)
☆118Updated 5 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
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- Code for "Efficient optimization of loops and limits with randomized telescoping sums"☆27Updated 5 years ago
- [NeurIPS 2019 Spotlight] High dimensional mean estimation and outlier detection in nearly-linear time.☆26Updated 5 years ago
- ☆214Updated 3 years ago
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- Python implementation of smooth optimal transport.☆58Updated 4 years ago
- Code for fast dpsgd implementations in JAX/TF☆59Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- ☆33Updated 4 years ago
- A library for exploring persistent homology☆100Updated 2 years ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated last year
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆106Updated 4 years ago
- A manifold optimization library for deep learning☆245Updated 3 years ago
- Official implementation (learning part) for paper: Numerically Accurate Hyperbolic Embeddings Using Tiling-Based Models☆18Updated 9 months ago
- Neural LSH [ICLR 2020] - Using supervised learning to produce better space partitions for fast nearest neighbor search.☆73Updated 4 years ago
- ☆99Updated 3 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 6 months ago
- Neural network verification in JAX☆142Updated last year
- ☆36Updated 3 years ago
- Hessian spectral density estimation in TF and Jax☆123Updated 4 years ago
- Functional TensorFlow Implementation of Singular Value Decomposition for paper Fast Graph Learning☆13Updated 4 years ago
- Cyanure: An Open-Source Toolbox for Empirical Risk Minimization☆64Updated last month
- MMD, Hausdorff and Sinkhorn divergences scaled up to 1,000,000 samples.☆55Updated 6 years ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- ICML 2020 Paper: Latent Variable Modelling with Hyperbolic Normalizing Flows☆53Updated 2 years ago
- Discrete Object Generation with Reversible Inductive Construction (NeurIPS 2019)☆30Updated 4 years ago
- ☆103Updated 10 months ago