EigenPro / EigenPro3
☆19Updated 7 months ago
Related projects: ⓘ
- IVON optimizer for neural networks based on variational learning.☆45Updated last month
- Code for the paper: "Independent mechanism analysis, a new concept?"☆25Updated last year
- Sketched matrix decompositions for PyTorch☆64Updated 4 months ago
- EigenPro Iteration in PyTorch☆19Updated 8 months ago
- Hessian spectral density estimation in TF and Jax☆112Updated 4 years ago
- ☆29Updated 3 years ago
- ☆56Updated 3 years ago
- Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J☆63Updated 5 months ago
- Transformers with doubly stochastic attention☆40Updated 2 years ago
- Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature☆93Updated last month
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆93Updated 4 years ago
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆31Updated last year
- scipy linear operators for the Hessian, Fisher/GGN, and more in PyTorch☆17Updated this week
- Agustinus' very opiniated publication-ready plotting library☆51Updated 2 weeks ago
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆22Updated 2 years ago
- ☆19Updated 4 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆68Updated last month
- ☆96Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆43Updated last year
- Implementation of the Gromov-Wasserstein distance to the setting of Unbalanced Optimal Transport☆43Updated last year
- Learning the optimal transport map via input convex neural neworks☆41Updated 4 years ago
- A PyTorch re-implementation of "Doubly Reparameterized Gradient Estimators for Monte Carlo Objectives"☆18Updated 5 years ago
- Implementation of Action Matching☆35Updated last year
- This repository contains code for applying Riemannian geometry in machine learning.☆75Updated 3 years ago
- Lipschitz Neural Networks described in "Sorting Out Lipschitz Function Approximation" (ICML 2019).☆54Updated 4 years ago
- ☆36Updated 3 years ago
- Efficient empirical NTKs in PyTorch☆15Updated 2 years ago
- ☆16Updated 8 months ago
- ☆56Updated last year
- ☆49Updated last year