mctorch / mctorch
A manifold optimization library for deep learning
☆241Updated 3 years ago
Alternatives and similar repositories for mctorch:
Users that are interested in mctorch are comparing it to the libraries listed below
- Optimization with orthogonal constraints and on general manifolds☆126Updated 4 years ago
- hessian in pytorch☆186Updated 4 years ago
- A PyTorch library for two-sample tests☆237Updated last year
- Hypergradient descent☆143Updated 7 months ago
- Code for "Functional variational Bayesian neural networks" (https://arxiv.org/abs/1903.05779)☆80Updated 4 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆142Updated last year
- ☆261Updated last year
- MMD, Hausdorff and Sinkhorn divergences scaled up to 1,000,000 samples.☆54Updated 5 years ago
- Preconditioned Conjugate Gradient in Pytorch☆136Updated 11 months ago
- Pytorch implementation of Block Neural Autoregressive Flow☆177Updated 3 years ago
- Convolutional Neural Tangent Kernel☆108Updated 5 years ago
- Code for ICML 2018 paper on "Fast and Scalable Bayesian Deep Learning by Weight-Perturbation in Adam" by Khan, Nielsen, Tangkaratt, Lin, …☆110Updated 6 years ago
- PyTorch implementation of Neural Processes☆88Updated 5 years ago
- Real NVP PyTorch a Minimal Working Example | Normalizing Flow☆137Updated 4 years ago
- ☆178Updated 5 years ago
- Pytorch implementation of Neural Processes for functions and images☆228Updated 2 years ago
- Deep neural network kernel for Gaussian process☆202Updated 4 years ago
- Sample code for running deterministic variational inference to train Bayesian neural networks☆100Updated 6 years ago
- code for "Residual Flows for Invertible Generative Modeling".☆265Updated last year
- ☆163Updated 5 months ago
- Understanding normalizing flows☆131Updated 5 years ago
- Toolbox to integrate optimal transport loss functions using automatic differentiation and Sinkhorn's algorithm☆437Updated 6 years ago
- Manifold-learning flows (ℳ-flows)☆229Updated 4 years ago
- Testing Nerual Tangent Kernel (NTK) on small UCI datasets☆83Updated 5 years ago
- Geometric loss functions between point clouds, images and volumes☆609Updated 11 months ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆140Updated last year
- Official pytorch implementation of the paper "Bayesian Meta-Learning for the Few-Shot Setting via Deep Kernels" (NeurIPS 2020)☆202Updated 2 years ago
- Lua implementation of Entropy-SGD☆81Updated 6 years ago
- Discrete Normalizing Flows implemented in PyTorch☆108Updated 3 years ago
- paper lists and information on mean-field theory of deep learning☆75Updated 5 years ago