g-benton / hessian-eff-dim
Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited
☆37Updated 2 years ago
Alternatives and similar repositories for hessian-eff-dim:
Users that are interested in hessian-eff-dim are comparing it to the libraries listed below
- Geometric Certifications of Neural Nets☆41Updated 2 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆47Updated 5 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- Large-batch Training, Neural Network Optimization☆9Updated 5 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- An empirical investigation of deep learning theory☆16Updated 5 years ago
- Limitations of the Empirical Fisher Approximation☆47Updated 2 months ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆14Updated 6 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Updated 5 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 4 years ago
- Implementation of Methods Proposed in Preventing Gradient Attenuation in Lipschitz Constrained Convolutional Networks (NeurIPS 2019)☆34Updated 4 years ago
- ☆30Updated 4 years ago
- ☆49Updated 4 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆13Updated 5 years ago
- ☆53Updated 9 months ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Updated 2 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Pytorch optimizers implementing Hilbert Constrained Gradient Descent☆19Updated 5 years ago
- Reliable Uncertainty Estimates in Deep Neural Networks using Noise Contrastive Priors☆62Updated 5 years ago
- Computing various norms/measures on over-parametrized neural networks☆49Updated 6 years ago
- Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning☆33Updated 5 years ago
- Code release for the ICLR paper☆20Updated 6 years ago
- Implementation of the Functional Neural Process models☆43Updated 4 years ago
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆17Updated 5 years ago
- Code for "Accelerating Natural Gradient with Higher-Order Invariance"☆30Updated 5 years ago
- ☆32Updated 6 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- ☆15Updated 5 years ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆53Updated 5 years ago