EigenPro / EigenPro-pytorch
EigenPro Iteration in PyTorch
☆19Updated 8 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
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆68Updated last month
- Transformers with doubly stochastic attention☆40Updated 2 years ago
- Discrete Normalizing Flows implemented in PyTorch☆105Updated 2 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆204Updated last month
- scipy linear operators for the Hessian, Fisher/GGN, and more in PyTorch☆17Updated this week
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆125Updated 5 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆106Updated this week
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆73Updated last year
- Official repository for "Categorical Normalizing Flows via Continuous Transformations"☆55Updated 3 years ago
- Sketched matrix decompositions for PyTorch☆64Updated 4 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆105Updated this week
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆176Updated last month
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆139Updated last year
- The Modified Differential Multiplier Method (MDMM) for PyTorch☆50Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆43Updated last year
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆93Updated 4 years ago
- Pytorch implementation of the Power Spherical distribution☆73Updated last month
- ☆215Updated 2 years ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆86Updated 5 months ago
- ☆19Updated 7 months ago
- Code for reproducing results in the sliced score matching paper (UAI 2019)☆138Updated 4 years ago
- Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"☆283Updated 3 years ago
- Regularized Neural ODEs (RNODE)☆81Updated 3 years ago
- Hessian spectral density estimation in TF and Jax☆112Updated 4 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆35Updated last year
- Large-scale, multi-GPU capable, kernel solver☆179Updated 2 months ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆68Updated 2 years ago
- Algorithms for computations on random manifolds made easier☆82Updated 9 months ago