Hessian trace estimation using PyTorch and Hutch++
☆20Oct 29, 2020Updated 5 years ago
Alternatives and similar repositories for hessian_trace_estimation
Users that are interested in hessian_trace_estimation are comparing it to the libraries listed below
Sorting:
- Neural Fixed-Point Acceleration for Convex Optimization☆29Oct 6, 2022Updated 3 years ago
- Official repository for the paper: "Trees with Attention for Set Prediction Tasks" (ICML21)☆10Jan 19, 2022Updated 4 years ago
- ☆13Nov 26, 2023Updated 2 years ago
- Quasi-Newton Algorithm for Stochastic Optimization☆11May 20, 2022Updated 3 years ago
- [NeurIPS 2020] Official Implementation: "SMYRF: Efficient Attention using Asymmetric Clustering".☆50Sep 6, 2023Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆29Sep 25, 2021Updated 4 years ago
- [ICLR 2025] Official PyTorch implementation of our paper for general continual learning "Advancing Prompt-Based Methods for Replay-Indepe…☆16Dec 21, 2025Updated 2 months ago
- Gradient-based Hyperparameter Optimization Over Long Horizons☆14Sep 29, 2021Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- Path integral based convolution and pooling☆30Feb 16, 2023Updated 3 years ago
- PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization☆16Apr 27, 2020Updated 5 years ago
- ☆126Jun 3, 2024Updated last year
- The official PyTorch implementation of CVPR2025 paper "Language Guided Concept Bottleneck Models for Interpretable Continual Learning"☆32Jun 17, 2025Updated 8 months ago
- A LinearOperator implementation for PyTorch☆18Jan 15, 2021Updated 5 years ago
- Contains code for the NeurIPS 2020 paper by Pan et al., "Continual Deep Learning by FunctionalRegularisation of Memorable Past"☆44Nov 10, 2020Updated 5 years ago
- Source code for ICLR 2020 paper: "Learning to Guide Random Search"☆40Sep 2, 2024Updated last year
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆126Jul 6, 2023Updated 2 years ago
- ☆22Feb 4, 2025Updated last year
- Test pytorch code with minimal computational overhead☆26Jun 8, 2023Updated 2 years ago
- ☆22Dec 3, 2021Updated 4 years ago
- The Differentiable Cross-Entropy Method☆125Aug 14, 2020Updated 5 years ago
- ☆21Jul 4, 2019Updated 6 years ago
- Implementation of REBAR in PyTorch☆17Jul 18, 2018Updated 7 years ago
- ☆50Oct 22, 2020Updated 5 years ago
- Official Release of "Learning the Stein Discrepancy for Training and Evaluating Energy-Based Models without Sampling"☆49Jul 8, 2020Updated 5 years ago
- The Official Implementation of Domain Adaptive Imitation Learning (DAIL)☆24Oct 26, 2020Updated 5 years ago
- A basic implementation of the paper Eigengame : PCA as a Nash Equilibrium☆21Jun 7, 2021Updated 4 years ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆53Mar 8, 2021Updated 4 years ago
- Official implementation of NeurIPS'21: Implicit SVD for Graph Representation Learning☆21Nov 4, 2021Updated 4 years ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Oct 25, 2021Updated 4 years ago
- [ICLR 2025] Bootstrapped Model Predictive Control☆33Jul 21, 2025Updated 7 months ago
- ☆25May 20, 2020Updated 5 years ago
- ☆22May 16, 2022Updated 3 years ago
- Unofficial PyTorch implementation of the paper "Conditional Channel Gated Networks for Task-Aware Continual Learning"☆20Jan 22, 2021Updated 5 years ago
- This repository implements the paper "Effective Training of Convolutional Neural Networks with Low-bitwidth Weights and Activations"☆20Aug 30, 2021Updated 4 years ago
- Code for NeurIPS 2020 Paper --- Continual Learning of a Mixed Sequence of Similar and Dissimilar Tasks☆21Oct 24, 2022Updated 3 years ago
- Meta Optimal Transport☆107May 25, 2023Updated 2 years ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆34Jun 17, 2022Updated 3 years ago
- code for paper "Large Language Models as End-to-end Combinatorial Optimization Solvers"☆59Oct 24, 2025Updated 4 months ago