Alternatives and similar repositories for pytorch-sso

Users that are interested in pytorch-sso are comparing it to the libraries listed below

• team-approx-bayes / dl-with-bayes

  View on GitHub
  Contains code for the NeurIPS 2019 paper "Practical Deep Learning with Bayesian Principles"
  ☆247Dec 3, 2019Updated 6 years ago
• cybertronai / autograd-lib

  View on GitHub
  ☆33May 21, 2020Updated 5 years ago
• Thrandis / EKFAC-pytorch

  View on GitHub
  Repository containing Pytorch code for EKFAC and K-FAC perconditioners.
  ☆152Jun 22, 2023Updated 2 years ago
• amirgholami / adahessian

  View on GitHub
  ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning
  ☆283Feb 27, 2023Updated 2 years ago
• fKunstner / limitations-empirical-fisher

  View on GitHub
  Limitations of the Empirical Fisher Approximation
  ☆49Mar 3, 2025Updated 11 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
• amirgholami / PyHessian

  View on GitHub
  PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks
  ☆773Jul 10, 2025Updated 7 months ago
• noahgolmant / pytorch-hessian-eigenthings

  View on GitHub
  Efficient PyTorch Hessian eigendecomposition tools!
  ☆384Feb 29, 2024Updated last year
• gpauloski / kfac-pytorch

  View on GitHub
  Distributed K-FAC preconditioner for PyTorch
  ☆95Updated this week
• kazukiosawa / asdl

  View on GitHub
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆191Dec 5, 2024Updated last year
• alecwangcq / KFAC-Pytorch

  View on GitHub
  Pytorch implementation of KFAC and E-KFAC (Natural Gradient).
  ☆132Jul 2, 2019Updated 6 years ago
• rgrosse / csc2541_examples

  View on GitHub
  ☆30Feb 11, 2021Updated 5 years ago
• gd-zhang / Weight-Decay

  View on GitHub
  Regularization, Neural Network Training Dynamics
  ☆14Jan 13, 2020Updated 6 years ago
• f-dangel / hbp

  View on GitHub
  Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations
  ☆21Mar 25, 2023Updated 2 years ago
• cybertronai / autograd-hacks

  View on GitHub
  ☆159Jun 8, 2022Updated 3 years ago
• renyiryry / kbfgs_neurips2020_public

  View on GitHub
  ☆23Mar 1, 2022Updated 3 years ago
• jotaf98 / pytorch-curveball

  View on GitHub
  A second-order optimizer for deep networks
  ☆25Oct 31, 2019Updated 6 years ago
• toiaydcdyywlhzvlob / backpack

  View on GitHub
  This repository is no longer maintained. Check
  ☆81Apr 23, 2020Updated 5 years ago
• team-approx-bayes / fromp

  View on GitHub
  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
• hjmshi / PyTorch-LBFGS

  View on GitHub
  A PyTorch implementation of L-BFGS.
  ☆625Feb 21, 2023Updated 2 years ago
• gd-zhang / noisy-K-FAC

  View on GitHub
  Natural Gradient, Variational Inference
  ☆29Jan 13, 2020Updated 6 years ago
• SebFar / radial_bnn

  View on GitHub
  Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning
  ☆33Apr 16, 2020Updated 5 years ago
• gd-zhang / Follow-the-Ridge

  View on GitHub
  Minimax Optimization, Stackelberg Games, Generative Adversarial Networks
  ☆19Feb 14, 2020Updated 6 years ago
• tfjgeorge / nngeometry

  View on GitHub
  {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
  ☆221Updated this week
• tensorflow / kfac

  View on GitHub
  An implementation of KFAC for TensorFlow
  ☆199Feb 11, 2022Updated 4 years ago
• aleximmer / Laplace

  View on GitHub
  Laplace approximations for Deep Learning.
  ☆534Apr 22, 2025Updated 9 months ago
• wjmaddox / drbayes

  View on GitHub
  Code Repo for "Subspace Inference for Bayesian Deep Learning"
  ☆83Jun 17, 2024Updated last year
• tyohei / chainerkfac

  View on GitHub
  A Chainer extension for K-FAC
  ☆20Jun 16, 2019Updated 6 years ago
• Lifelong-ML / LASEM

  View on GitHub
  Code for the ICML 2021 paper "Sharing Less is More: Lifelong Learning in Deep Networks with Selective Layer Transfer"
  ☆12Aug 17, 2021Updated 4 years ago
• emtiyaz / vadam

  View on GitHub
  Code for ICML 2018 paper on "Fast and Scalable Bayesian Deep Learning by Weight-Perturbation in Adam" by Khan, Nielsen, Tangkaratt, Lin, …
  ☆112Dec 19, 2018Updated 7 years ago
• pomonam / NoisyNaturalGradient

  View on GitHub
  TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".
  ☆60Jan 12, 2019Updated 7 years ago
• ssydasheng / FBNN

  View on GitHub
  Code for "Functional variational Bayesian neural networks" (https://arxiv.org/abs/1903.05779)
  ☆86Jul 19, 2020Updated 5 years ago
• wjmaddox / swa_gaussian

  View on GitHub
  Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"
  ☆478Jul 6, 2023Updated 2 years ago
• g-benton / hessian-eff-dim

  View on GitHub
  Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited
  ☆37Dec 27, 2022Updated 3 years ago
• tml-epfl / understanding-sam

  View on GitHub
  Towards Understanding Sharpness-Aware Minimization [ICML 2022]
  ☆38Jun 14, 2022Updated 3 years ago
• mariogeiger / hessian

  View on GitHub
  hessian in pytorch
  ☆187Oct 27, 2020Updated 5 years ago
• ratschlab / bnn_priors

  View on GitHub
  Code for the paper "Bayesian Neural Network Priors Revisited"
  ☆60Jul 1, 2021Updated 4 years ago
• tajanthan / pmf

  View on GitHub
  Proximal Mean-field for Neural Network Quantization
  ☆21Apr 9, 2020Updated 5 years ago
• team-approx-bayes / BayesBiNN

  View on GitHub
  Code for the paper "Training Binary Neural Networks with Bayesian Learning Rule
  ☆41Jan 4, 2022Updated 4 years ago