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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Measurements of Three-Level Hierarchical Structure in the Outliers in the Spectrum of Deepnet Hessians (ICML 2019)☆16Apr 27, 2019Updated 7 years ago
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆19May 19, 2019Updated 7 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆473Updated this week
- ☆31Feb 11, 2021Updated 5 years ago
- Full loss Hessian spectrum approximation tool.☆13Oct 9, 2019Updated 6 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- ☆85Jan 15, 2020Updated 6 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,386Mar 1, 2024Updated 2 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Dec 27, 2022Updated 3 years ago
- can calculate the Hessian matrix and/or its spectrum for simple neural nets☆11May 7, 2018Updated 8 years ago
- Code accompanying our paper "Finding trainable sparse networks through Neural Tangent Transfer" to be published at ICML-2020.☆13Jun 14, 2020Updated 6 years ago
- Matlab code implementing Minimum Probability Flow Learning.☆69Sep 17, 2014Updated 11 years ago
- MDL Complexity computations and experiments from the paper "Revisiting complexity and the bias-variance tradeoff".☆18Jun 12, 2023Updated 3 years ago
- ☆16Jun 3, 2025Updated last year
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Jul 4, 2022Updated 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- codebase for "A Theory of the Inductive Bias and Generalization of Kernel Regression and Wide Neural Networks"☆52May 2, 2023Updated 3 years ago
- ☆12Oct 8, 2020Updated 5 years ago
- Energy-Aware Neural Architecture Optimization with Fast Splitting Steepest Descent☆14Feb 6, 2020Updated 6 years ago
- Forward mode laplacian implemented in JAX tracer☆32Jan 7, 2026Updated 7 months ago
- [ICML'21 Oral] Improving Lossless Compression Rates via Monte Carlo Bits-Back Coding☆14Jun 10, 2021Updated 5 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 7 years ago
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆30Nov 26, 2022Updated 3 years ago
- Matlab Algorithms for Randomized Linear Algebra☆16Jun 30, 2023Updated 3 years ago
- ☆331Sep 20, 2022Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 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
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆615Jul 17, 2026Updated 3 weeks ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆22Mar 25, 2023Updated 3 years ago
- Code release for the ICLR paper☆22Jun 13, 2018Updated 8 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆135Jul 2, 2019Updated 7 years ago
- ☆28Oct 21, 2022Updated 3 years ago
- ICML 2020, Estimating Generalization under Distribution Shifts via Domain-Invariant Representations☆23Jun 30, 2020Updated 6 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆330Updated this week
- Probabilistic Solution of Differential Equations☆13Jun 19, 2022Updated 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Sampled Quasi-Newton Methods for Deep Learning☆21May 21, 2020Updated 6 years ago
- ☆12Sep 26, 2019Updated 6 years ago
- code to reproduce the empirical results in the research paper☆40Oct 12, 2021Updated 4 years ago
- code for "Residual Flows for Invertible Generative Modeling".☆275Sep 19, 2023Updated 2 years ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,056Jun 4, 2026Updated 2 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆106Oct 23, 2023Updated 2 years ago
- Path Tracing in JAX☆75Dec 15, 2019Updated 6 years ago