ilyaraz / ot_estimators
Implementations of several fast approximate algorithms for geometric optimal transport (OT)
☆118Updated 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
- ☆33Updated 4 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆106Updated 4 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- Python implementation of projection losses.☆25Updated 5 years ago
- ☆99Updated 3 years ago
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…☆240Updated 4 years ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated last year
- A library for exploring persistent homology☆100Updated 2 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 5 months ago
- Code for fast dpsgd implementations in JAX/TF☆59Updated 2 years ago
- ☆214Updated 3 years ago
- A study of performance of optimal transport.☆10Updated 4 years ago
- Statistical adaptive stochastic optimization methods☆33Updated 5 years ago
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- Persistent Homology Algorithm Toolbox (PHAT).☆35Updated 5 years ago
- A manifold optimization library for deep learning☆242Updated 3 years ago
- Python implementation of smooth optimal transport.☆57Updated 3 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Neural network verification in JAX☆141Updated last year
- ☆101Updated 9 months ago
- Differentiable and numerically stable implementation of the matrix exponential☆33Updated 4 years ago
- Loss Landscapes of Regularized Linear Autoencoders☆144Updated 2 years ago
- Matlab code for tree-Wasserstein distance in the paper "Tree-Sliced Variants of Wasserstein Distances", NeurIPS, 2019. (Tam Le, Makoto Y…☆11Updated 5 years ago
- Fast optimal transport code☆56Updated 4 years ago
- Efficient LSH-based kernel density estimation☆29Updated 5 years ago
- PAC-Bayes with Backprop - Tighter risk certificates for neural networks☆24Updated 3 years ago
- Optimization with orthogonal constraints and on general manifolds☆128Updated 4 years ago