Alternatives and similar repositories for hessian_trace_estimation

Users that are interested in hessian_trace_estimation are comparing it to the libraries listed below

• facebookresearch / neural-scs

  View on GitHub
  Neural Fixed-Point Acceleration for Convex Optimization
  ☆29Oct 6, 2022Updated 3 years ago
• apple / ml-agm

  View on GitHub
  ☆44May 3, 2024Updated last year
• TAU-MLwell / Set-Tree

  View on GitHub
  Official repository for the paper: "Trees with Attention for Set Prediction Tasks" (ICML21)
  ☆10Jan 19, 2022Updated 4 years ago
• giannisdaras / smyrf

  View on GitHub
  [NeurIPS 2020] Official Implementation: "SMYRF: Efficient Attention using Asymmetric Clustering".
  ☆50Sep 6, 2023Updated 2 years ago
• ermongroup / fast_feedforward_computation

  View on GitHub
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆29Sep 25, 2021Updated 4 years ago
• mzf666 / LORO-main

  View on GitHub
  Official implementation of ICLR 2025 'LORO: Parameter and Memory Efficient Pretraining via Low-rank Riemannian Optimization'
  ☆16Apr 24, 2025Updated 9 months ago
• polo5 / FDS

  View on GitHub
  Gradient-based Hyperparameter Optimization Over Long Horizons
  ☆14Sep 29, 2021Updated 4 years ago
• bangxiangyong / baetorch

  View on GitHub
  Python library for Bayesian Autoencoders
  ☆13Jun 10, 2022Updated 3 years ago
• kangzhiq / MISA

  View on GitHub
  [ICLR 2025] Official PyTorch implementation of our paper for general continual learning "Advancing Prompt-Based Methods for Replay-Indepe…
  ☆16Dec 21, 2025Updated last month
• YuGuangWang / PAN

  View on GitHub
  Path integral based convolution and pooling
  ☆30Feb 16, 2023Updated 2 years ago
• prolearner / hypertorch

  View on GitHub
  ☆126Jun 3, 2024Updated last year
• CUAI / Neural-Manifold-Ordinary-Differential-Equations

  View on GitHub
  [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)
  ☆125Jul 6, 2023Updated 2 years ago
• take-koshizuka / NLSB

  View on GitHub
  ☆22Feb 4, 2025Updated last year
• mfinzi / residual-pathway-priors

  View on GitHub
  ☆22Dec 3, 2021Updated 4 years ago
• facebookresearch / dcem

  View on GitHub
  The Differentiable Cross-Entropy Method
  ☆124Aug 14, 2020Updated 5 years ago
• mgallieriac / LASSO-MPC

  View on GitHub
  PhD Publications and Thesis on LASSO Model Predictive Control
  ☆20Jun 2, 2019Updated 6 years ago
• Junyoungpark / pyjssp

  View on GitHub
  ☆21Jul 4, 2019Updated 6 years ago
• ermongroup / dail

  View on GitHub
  The Official Implementation of Domain Adaptive Imitation Learning (DAIL)
  ☆24Oct 26, 2020Updated 5 years ago
• gauravdhama / eigengame_deepmind

  View on GitHub
  A basic implementation of the paper Eigengame : PCA as a Nash Equilibrium
  ☆21Jun 7, 2021Updated 4 years ago
• IST-DASLab / WoodFisher

  View on GitHub
  Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)
  ☆53Mar 8, 2021Updated 4 years ago
• wertyuilife2 / bmpc

  View on GitHub
  [ICLR 2025] Bootstrapped Model Predictive Control
  ☆32Jul 21, 2025Updated 6 months ago
• a7b23 / H-Divergence

  View on GitHub
  ☆22May 16, 2022Updated 3 years ago
• lit-leo / cgate

  View on GitHub
  Unofficial PyTorch implementation of the paper "Conditional Channel Gated Networks for Task-Aware Continual Learning"
  ☆20Jan 22, 2021Updated 5 years ago
• katalinic / sdflows

  View on GitHub
  JAX exponential map normalising flows on sphere
  ☆17Oct 4, 2020Updated 5 years ago
• ZhengyaoJiang / GTG

  View on GitHub
  Source code of "Grid-to-Graph: Flexible Spatial Relational Inductive Biases for Reinforcement Learning" (AAMAS 2021).
  ☆29Aug 19, 2021Updated 4 years ago
• ZixuanKe / CAT

  View on GitHub
  Code for NeurIPS 2020 Paper --- Continual Learning of a Mixed Sequence of Similar and Dissimilar Tasks
  ☆21Oct 24, 2022Updated 3 years ago
• facebookresearch / meta-ot

  View on GitHub
  Meta Optimal Transport
  ☆107May 25, 2023Updated 2 years ago
• jaehong31 / APD

  View on GitHub
  "Scalable and Order-robust Continual Learning with Additive Parameter Decomposition", ICLR 2020
  ☆23May 25, 2022Updated 3 years ago
• duvenaud / jaxde

  View on GitHub
  Prototypes of differentiable differential equation solvers in JAX.
  ☆27Feb 3, 2020Updated 6 years ago
• younggyoseo / RE3

  View on GitHub
  RE3: State Entropy Maximization with Random Encoders for Efficient Exploration
  ☆69Jul 29, 2021Updated 4 years ago
• locuslab / mdeq

  View on GitHub
  [NeurIPS'20] Multiscale Deep Equilibrium Models
  ☆237Jun 29, 2021Updated 4 years ago
• aravindr93 / trajopt

  View on GitHub
  Trajectory optimization algorithms for robotic control.
  ☆118Nov 20, 2022Updated 3 years ago
• linghan1997 / Regression-based-Analytic-Incremental-Learning

  View on GitHub
  ☆32Feb 24, 2025Updated 11 months ago
• linhaowei1 / TPL

  View on GitHub
  ✌[ICLR 2024] Class Incremental Learning via Likelihood Ratio Based Task Prediction
  ☆32Oct 29, 2024Updated last year
• google / deluca

  View on GitHub
  Performant, differentiable reinforcement learning
  ☆125Aug 2, 2025Updated 6 months ago
• f-dangel / backpack

  View on GitHub
  BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
  ☆604Nov 28, 2025Updated 2 months ago
• alecwangcq / EigenDamage-Pytorch

  View on GitHub
  Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934
  ☆113Mar 3, 2020Updated 5 years ago
• ericjang / pt-jax

  View on GitHub
  Path Tracing in JAX
  ☆73Dec 15, 2019Updated 6 years ago
• gehring / fax

  View on GitHub
  ☆80Sep 4, 2021Updated 4 years ago