btaba / sinkhorn_knopp
python implementation of Sinkhorn-Knopp
☆33Updated 6 years ago
Alternatives and similar repositories for sinkhorn_knopp:
Users that are interested in sinkhorn_knopp are comparing it to the libraries listed below
- Implementation of the Gromov-Wasserstein distance to the setting of Unbalanced Optimal Transport☆43Updated last year
- Code for Sliced Gromov-Wasserstein☆66Updated 5 years ago
- ☆53Updated 6 months ago
- Source code for the ICML2019 paper "Subspace Robust Wasserstein Distances"☆28Updated 5 years ago
- Contains the code relative to the paper Partial Gromov-Wasserstein with Applications on Positive-Unlabeled Learning https://arxiv.org/abs…☆21Updated 4 years ago
- Stochastic algorithms for computing Regularized Optimal Transport☆57Updated 6 years ago
- Python implementation of smooth optimal transport.☆57Updated 3 years ago
- ☆58Updated last year
- Learning Autoencoders with Relational Regularization☆45Updated 4 years ago
- Code for "Depth Uncertainty in Neural Networks" (https://arxiv.org/abs/2006.08437)☆72Updated last year
- implements optimal transport algorithms in pytorch☆93Updated 2 years ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆54Updated 3 years ago
- PyTorch implementation of Stein Variational Gradient Descent☆42Updated last year
- ☆40Updated 5 years ago
- ☆31Updated 4 years ago
- It is a repo which allows to compute all divergences derived from the theory of entropically regularized, unbalanced optimal transport. I…☆28Updated 2 years ago
- ☆64Updated 4 years ago
- [NeurIPS 2020] Coresets for Robust Training of Neural Networks against Noisy Labels☆32Updated 3 years ago
- Code to accompany paper 'Bayesian Deep Ensembles via the Neural Tangent Kernel'☆27Updated 3 years ago
- Official Release of "Learning the Stein Discrepancy for Training and Evaluating Energy-Based Models without Sampling"☆49Updated 4 years ago
- ☆36Updated 3 years ago
- Code for experiments to learn uncertainty☆30Updated last year
- Implementation of the Sliced Wasserstein Autoencoder using PyTorch☆101Updated 6 years ago
- MMD, Hausdorff and Sinkhorn divergences scaled up to 1,000,000 samples.☆54Updated 5 years ago
- Official implementation of "Intrinsic Dimension, Persistent Homology and Generalization in Neural Networks", NeurIPS 2021.☆31Updated 3 years ago
- Code for "On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertainty".☆110Updated 2 years ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Updated 2 years ago
- Computing various measures and generalization bounds on convolutional and fully connected networks☆35Updated 6 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated last year
- Gaussian Process Prior Variational Autoencoder☆81Updated 6 years ago