ilyaraz / ot_estimators
Implementations of several fast approximate algorithms for geometric optimal transport (OT)
☆118Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for ot_estimators
- 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
- Optimization with orthogonal constraints and on general manifolds☆126Updated 4 years ago
- Python implementation of smooth optimal transport.☆56Updated 3 years ago
- A library for exploring persistent homology☆101Updated last year
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆76Updated 8 months ago
- Pytorch implementation of the Power Spherical distribution☆73Updated 4 months ago
- ☆213Updated 2 years ago
- Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.☆134Updated last year
- Differentiable and numerically stable implementation of the matrix exponential☆32Updated 4 years ago
- ☆97Updated 2 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Matlab code for tree-Wasserstein distance in the paper "Tree-Sliced Variants of Wasserstein Distances", NeurIPS, 2019. (Tam Le, Makoto Y…☆12Updated 4 years ago
- MMD, Hausdorff and Sinkhorn divergences scaled up to 1,000,000 samples.☆54Updated 5 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆142Updated last year
- Riemannian Convex Potential Maps☆68Updated last year
- Code for fast dpsgd implementations in JAX/TF☆58Updated 2 years ago
- Graduate topics course on learning discrete latent structure.☆66Updated 5 years ago
- Large-scale, multi-GPU capable, kernel solver☆181Updated 4 months ago
- Statistical adaptive stochastic optimization methods☆32Updated 4 years ago
- Python implementation of projection losses.☆25Updated 5 years ago
- A Python implementation of Monge optimal transportation☆48Updated last year
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆107Updated this week
- Neural LSH [ICLR 2020] - Using supervised learning to produce better space partitions for fast nearest neighbor search.☆71Updated 3 years ago
- ☆153Updated 2 years ago
- ☆99Updated 4 months ago
- Efficient LSH-based kernel density estimation☆28Updated 5 years ago
- Gabriel Peyré, Marco Cuturi, Justin Solomon, Gromov-Wasserstein Averaging of Kernel and Distance Matrices, Proc. of ICML 2016.☆73Updated 8 years ago
- B-Spline CNNs on Lie groups☆53Updated 4 years ago
- Differentiable bitonic sorting☆138Updated 4 years ago