Hessian spectral density estimation in TF and Jax
☆125Sep 6, 2020Updated 5 years ago
Alternatives and similar repositories for spectral-density
Users that are interested in spectral-density are comparing it to the libraries listed below
Sorting:
- Measurements of Three-Level Hierarchical Structure in the Outliers in the Spectrum of Deepnet Hessians (ICML 2019)☆16Apr 27, 2019Updated 6 years ago
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆19May 19, 2019Updated 6 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆386Feb 29, 2024Updated 2 years ago
- ☆30Feb 11, 2021Updated 5 years ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆777Jul 10, 2025Updated 7 months ago
- Fast and Easy Infinite Neural Networks in Python☆2,375Mar 1, 2024Updated 2 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Dec 27, 2022Updated 3 years ago
- ☆83Jan 15, 2020Updated 6 years ago
- Public code for a paper "Lipschitz-Margin Training: Scalable Certification of Perturbation Invariance for Deep Neural Networks."☆35Dec 18, 2018Updated 7 years ago
- Code accompanying our paper "Finding trainable sparse networks through Neural Tangent Transfer" to be published at ICML-2020.☆13Jun 14, 2020Updated 5 years ago
- Energy-Aware Neural Architecture Optimization with Fast Splitting Steepest Descent☆14Feb 6, 2020Updated 6 years ago
- ☆16Jun 3, 2025Updated 8 months ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆606Nov 28, 2025Updated 3 months ago
- MetaC provides a read-eval-print loop (a REPL) and notebook interactive development environment (a NIDE) for C programming. MetaC also …☆12Feb 16, 2026Updated last week
- ☆100Dec 8, 2021Updated 4 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆133Jul 2, 2019Updated 6 years ago
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 5 years ago
- Supporting code for "Autoregressive neural-network wavefunctions for ab initio quantum chemistry".☆43Jun 17, 2022Updated 3 years ago
- PyTorch AutoNEB implementation to identify minimum energy paths, e.g. in neural network loss landscapes☆56Sep 14, 2022Updated 3 years ago
- Image augmentation library for Jax☆41Apr 9, 2024Updated last year
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- ☆44Dec 4, 2018Updated 7 years ago
- code for "Residual Flows for Invertible Generative Modeling".☆273Sep 19, 2023Updated 2 years ago
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆29Nov 26, 2022Updated 3 years ago
- Matlab code implementing Minimum Probability Flow Learning.☆70Sep 17, 2014Updated 11 years ago
- ☆620Jan 23, 2026Updated last month
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆21Mar 25, 2023Updated 2 years ago
- Code release for the ICLR paper☆21Jun 13, 2018Updated 7 years ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- ☆16Oct 2, 2022Updated 3 years ago
- variational free-energy of dense hydrogen☆14Sep 25, 2023Updated 2 years ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 6 months ago
- Official code for "Algorithmic Capabilities of Random Transformers" (NeurIPS 2024)☆16Sep 28, 2024Updated last year
- ☆12Oct 8, 2020Updated 5 years ago
- ☆17Nov 25, 2017Updated 8 years ago
- This repository contains the code used in a publication 'Active Learning for Decision-Making from Imbalanced Observational Data', Iiris S…☆11May 14, 2019Updated 6 years ago
- Convolutional Neural Tangent Kernel☆112Oct 22, 2019Updated 6 years ago
- ☆328Sep 20, 2022Updated 3 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Mar 7, 2023Updated 2 years ago