amirgholami / PyHessian

Related projects ⓘ

Alternatives and complementary repositories for PyHessian

• noahgolmant / pytorch-hessian-eigenthings
  Efficient PyTorch Hessian eigendecomposition tools!
  ☆363Updated 8 months ago
• f-dangel / backpack
  BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
  ☆561Updated 6 months ago
• amirgholami / adahessian
  ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning
  ☆265Updated last year
• marcellodebernardi / loss-landscapes
  Approximating neural network loss landscapes in low-dimensional parameter subspaces for PyTorch
  ☆301Updated 11 months ago
• facebookresearch / open_lth
  A repository in preparation for open-sourcing lottery ticket hypothesis code.
  ☆626Updated 2 years ago
• timgaripov / dnn-mode-connectivity
  Mode Connectivity and Fast Geometric Ensembles in PyTorch
  ☆265Updated 2 years ago
• ganguli-lab / Synaptic-Flow
  ☆218Updated 3 months ago
• JJGO / shrinkbench
  PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.
  ☆423Updated last year
• tfjgeorge / nngeometry
  {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
  ☆206Updated last month
• google-research / rigl
  End-to-end training of sparse deep neural networks with little-to-no performance loss.
  ☆317Updated last year
• rahulvigneswaran / Lottery-Ticket-Hypothesis-in-Pytorch
  This repository contains a Pytorch implementation of the paper "The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks"…
  ☆325Updated last year
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,394Updated this week
• cybertronai / autograd-hacks
  ☆153Updated 2 years ago
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆468Updated last month
• google-research / sam
  ☆565Updated 2 weeks ago
• allenai / hidden-networks
  ☆184Updated 3 years ago
• hjmshi / PyTorch-LBFGS
  A PyTorch implementation of L-BFGS.
  ☆586Updated last year
• BayesWatch / nas-without-training
  Code for Neural Architecture Search without Training (ICML 2021)
  ☆460Updated 3 years ago
• lucaslie / torchprune
  A research library for pytorch-based neural network pruning, compression, and more.
  ☆160Updated last year
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆657Updated 2 years ago
• AntixK / PyTorch-Model-Compare
  Compare neural networks by their feature similarity
  ☆342Updated last year
• mariogeiger / hessian
  hessian in pytorch
  ☆185Updated 4 years ago
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆178Updated 2 months ago
• locuslab / deq
  [NeurIPS'19] Deep Equilibrium Models
  ☆727Updated 2 years ago
• 1adrianb / pytorch-estimate-flops
  Estimate/count FLOPS for a given neural network using pytorch
  ☆304Updated 2 years ago
• TimDettmers / sparse_learning
  Sparse learning library and sparse momentum resources.
  ☆378Updated 2 years ago
• facebookresearch / higher
  higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…
  ☆1,589Updated 2 years ago
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆474Updated 2 years ago
• wjmaddox / swa_gaussian
  Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"
  ☆451Updated last year
• dfdazac / wassdistance
  Approximating Wasserstein distances with PyTorch
  ☆455Updated last year