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:
- 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
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆778Jul 10, 2025Updated 8 months ago
- ☆30Feb 11, 2021Updated 5 years ago
- Full loss Hessian spectrum approximation tool.☆13Oct 9, 2019Updated 6 years ago
- ☆83Jan 15, 2020Updated 6 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,377Mar 1, 2024Updated 2 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Dec 27, 2022Updated 3 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
- can calculate the Hessian matrix and/or its spectrum for simple neural nets☆11May 7, 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
- MDL Complexity computations and experiments from the paper "Revisiting complexity and the bias-variance tradeoff".☆18Jun 12, 2023Updated 2 years ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Jul 4, 2022Updated 3 years ago
- Forward mode laplacian implemented in JAX tracer☆30Jan 7, 2026Updated 2 months ago
- ☆12Oct 8, 2020Updated 5 years ago
- ☆100Dec 8, 2021Updated 4 years ago
- Energy-Aware Neural Architecture Optimization with Fast Splitting Steepest Descent☆14Feb 6, 2020Updated 6 years ago
- [ICML'21 Oral] Improving Lossless Compression Rates via Monte Carlo Bits-Back Coding☆14Jun 10, 2021Updated 4 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
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆29Nov 26, 2022Updated 3 years ago
- Official code for "Algorithmic Capabilities of Random Transformers" (NeurIPS 2024)☆16Sep 28, 2024Updated last year
- MetaC provides a read-eval-print loop (a REPL) and notebook interactive development environment (a NIDE) for C programming. MetaC also …☆12Mar 4, 2026Updated 2 weeks ago
- Matlab Algorithms for Randomized Linear Algebra☆15Jun 30, 2023Updated 2 years ago
- We define and estimate smooth unique information of samples with respect to classifier weights and predictions. We compute these quantiti…☆11Mar 9, 2021Updated 5 years ago
- PyTorch AutoNEB implementation to identify minimum energy paths, e.g. in neural network loss landscapes☆56Sep 14, 2022Updated 3 years ago
- Implementation of AlphaZero in PyTorch.☆10Apr 19, 2019Updated 6 years ago
- ☆327Sep 20, 2022Updated 3 years ago
- Code release for the ICLR paper☆21Jun 13, 2018Updated 7 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆606Nov 28, 2025Updated 3 months ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆21Mar 25, 2023Updated 2 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆133Jul 2, 2019Updated 6 years ago
- ICML 2020, Estimating Generalization under Distribution Shifts via Domain-Invariant Representations☆23Jun 30, 2020Updated 5 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆317Updated this week
- Probabilistic Solution of Differential Equations☆13Jun 19, 2022Updated 3 years ago
- ☆12Sep 26, 2019Updated 6 years ago
- Scalable Computation of Hessian Diagonals☆14Jun 2, 2024Updated last year
- code to reproduce the empirical results in the research paper☆38Oct 12, 2021Updated 4 years ago
- pyhessian is a TensorFlow module which can be used to estimate Hessian matrices☆25Mar 26, 2021Updated 4 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago