Alternatives and similar repositories for backpack

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

• noahgolmant / pytorch-hessian-eigenthings
  Efficient PyTorch Hessian eigendecomposition tools!
  ☆374Updated last year
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆514Updated 2 months ago
• amirgholami / adahessian
  ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning
  ☆277Updated 2 years ago
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆687Updated 3 years ago
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆188Updated 7 months ago
• tfjgeorge / nngeometry
  {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
  ☆215Updated 3 weeks ago
• cybertronai / pytorch-sso
  PyTorch-SSO: Scalable Second-Order methods in PyTorch
  ☆147Updated last year
• getkeops / keops
  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
  ☆1,106Updated last week
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆480Updated 3 years ago
• AdamCobb / hamiltorch
  PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks
  ☆451Updated 10 months ago
• google-research / robustness_metrics
  ☆470Updated 2 months ago
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆305Updated 2 weeks ago
• wjmaddox / swa_gaussian
  Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"
  ☆468Updated 2 years ago
• amirgholami / PyHessian
  PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks
  ☆745Updated this week
• izmailovpavel / understandingbdl
  ☆240Updated 2 years ago
• cybertronai / autograd-hacks
  ☆157Updated 3 years ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,432Updated this week
• y0ast / slurm-for-ml
  A Machine Learning workflow for Slurm.
  ☆149Updated 4 years ago
• google-research / fast-soft-sort
  Fast Differentiable Sorting and Ranking
  ☆606Updated last year
• thegregyang / GP4A
  Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…
  ☆243Updated 4 years ago
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆123Updated 4 years ago
• wgrathwohl / JEM
  Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"
  ☆423Updated 2 years ago
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆979Updated 2 months ago
• SirRob1997 / Crowded-Valley---Results
  This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"
  ☆179Updated 3 years ago
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
  ☆614Updated this week
• locuslab / deq
  [NeurIPS'19] Deep Equilibrium Models
  ☆753Updated 3 years ago
• didriknielsen / survae_flows
  Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"
  ☆287Updated 4 years ago
• google-deepmind / distrax
  ☆580Updated last month
• Thrandis / EKFAC-pytorch
  Repository containing Pytorch code for EKFAC and K-FAC perconditioners.
  ☆143Updated 2 years ago
• rajatvd / NTK
  Code for experiments in my blog post on the Neural Tangent Kernel: https://eigentales.com/NTK
  ☆176Updated 5 years ago