kazukiosawa / asdlView external linksLinks
ASDL: Automatic Second-order Differentiation Library for PyTorch
☆191Dec 5, 2024Updated last year
Alternatives and similar repositories for asdl
Users that are interested in asdl are comparing it to the libraries listed below
Sorting:
- Laplace approximations for Deep Learning.☆534Apr 22, 2025Updated 9 months ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆63Updated this week
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆604Nov 28, 2025Updated 2 months ago
- A differentiation API for PyTorch☆30Jun 21, 2020Updated 5 years ago
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆24Nov 4, 2024Updated last year
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆148Oct 1, 2023Updated 2 years ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆21Mar 25, 2023Updated 2 years ago
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆10Oct 21, 2022Updated 3 years ago
- Official repository for the paper: "Trees with Attention for Set Prediction Tasks" (ICML21)☆10Jan 19, 2022Updated 4 years ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 2 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆311Feb 9, 2026Updated last week
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆15Nov 4, 2024Updated last year
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 5 years ago
- Official implementation of NeurIPS'21: Implicit SVD for Graph Representation Learning☆21Nov 4, 2021Updated 4 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Jul 2, 2019Updated 6 years ago
- [AAAI 2021 Workshop] The official repository for the LST-MAP model for few-shot image classification.☆13Feb 12, 2021Updated 5 years ago
- An adaptive training algorithm for residual network☆17Aug 22, 2020Updated 5 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆310Jan 12, 2026Updated last month
- Efficient PyTorch Hessian eigendecomposition tools!☆384Feb 29, 2024Updated last year
- Differentiable scientific computing library☆156Aug 14, 2024Updated last year
- Code for "Accelerating Natural Gradient with Higher-Order Invariance"☆30Jun 28, 2019Updated 6 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆44Jun 6, 2025Updated 8 months ago
- Groundstate Reconstructive Energies using Neural Autoregressive Distribution Estimators (GreNADE)☆10Apr 1, 2022Updated 3 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆190Jan 11, 2026Updated last month
- Constrained optimization toolkit for PyTorch☆706Jul 29, 2025Updated 6 months ago
- A Generic Tensor-Network library that is designed for quantum simulation, base on the pytorch☆59Jun 21, 2019Updated 6 years ago
- MPS in Julia minicourse☆19Oct 13, 2017Updated 8 years ago
- Differentiable programming on quantum circuits with Flux☆19Feb 12, 2020Updated 6 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆221Feb 9, 2026Updated last week
- ☆16Apr 26, 2023Updated 2 years ago
- ☆11Apr 5, 2021Updated 4 years ago
- This repository contains the source code for the VFK algorithm introduced in ArXiv:2204.03454☆16Jun 10, 2025Updated 8 months ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆91Oct 19, 2022Updated 3 years ago
- ☆31May 4, 2020Updated 5 years ago
- A basic implementation of the paper Eigengame : PCA as a Nash Equilibrium☆21Jun 7, 2021Updated 4 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆23Sep 9, 2024Updated last year
- Infinite order automatic differentiation for Monte Carlo with unnormalized probability distribution☆24Mar 24, 2023Updated 2 years ago
- Limitations of the Empirical Fisher Approximation☆49Mar 3, 2025Updated 11 months ago
- The differentiable OSQP solver layer for PyTorch☆65Dec 26, 2022Updated 3 years ago