f-dangel / backpack

Related projects ⓘ

Alternatives and complementary repositories for backpack

• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆468Updated last month
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆656Updated 2 years ago
• noahgolmant / pytorch-hessian-eigenthings
  Efficient PyTorch Hessian eigendecomposition tools!
  ☆362Updated 8 months ago
• AdamCobb / hamiltorch
  PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks
  ☆421Updated 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…
  ☆301Updated last week
• getkeops / keops
  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
  ☆1,054Updated this week
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆474Updated 2 years ago
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆178Updated 2 months ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,395Updated this week
• amirgholami / adahessian
  ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning
  ☆265Updated last year
• izmailovpavel / understandingbdl
  ☆233Updated last year
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆521Updated this week
• google-research / robustness_metrics
  ☆466Updated 3 months ago
• mariogeiger / hessian
  hessian in pytorch
  ☆185Updated 4 years ago
• amirgholami / PyHessian
  PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks
  ☆685Updated 6 months ago
• bayesiains / nflows
  Normalizing flows in PyTorch
  ☆845Updated 11 months ago
• wjmaddox / swa_gaussian
  Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"
  ☆451Updated last year
• tfjgeorge / nngeometry
  {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
  ☆206Updated last month
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆929Updated last month
• SirRob1997 / Crowded-Valley---Results
  This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"
  ☆180Updated 3 years ago
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆115Updated 4 years ago
• didriknielsen / survae_flows
  Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"
  ☆283Updated 3 years ago
• FalkonML / falkon
  Large-scale, multi-GPU capable, kernel solver
  ☆181Updated 3 months ago
• google-deepmind / distrax
  ☆535Updated last month
• cybertronai / pytorch-sso
  PyTorch-SSO: Scalable Second-Order methods in PyTorch
  ☆141Updated last year
• wgrathwohl / JEM
  Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"
  ☆417Updated 2 years ago
• cybertronai / autograd-hacks
  ☆153Updated 2 years ago
• google-research / ott
  ☆213Updated 2 years ago
• jeanfeydy / geomloss
  Geometric loss functions between point clouds, images and volumes
  ☆595Updated 9 months ago
• Thrandis / EKFAC-pytorch
  Repository containing Pytorch code for EKFAC and K-FAC perconditioners.
  ☆140Updated last year