Alternatives and similar repositories for loss-landscape-anim

Users that are interested in loss-landscape-anim are comparing it to the libraries listed below

• tfjgeorge / nngeometry
  {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
  ☆215Updated 2 months ago
• marcellodebernardi / loss-landscapes
  Approximating neural network loss landscapes in low-dimensional parameter subspaces for PyTorch
  ☆339Updated last year
• ppope / dimensions
  Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J
  ☆71Updated last year
• apple / learning-subspaces
  ☆133Updated 4 years ago
• timgaripov / dnn-mode-connectivity
  Mode Connectivity and Fast Geometric Ensembles in PyTorch
  ☆274Updated 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
• KellerJordan / REPAIR
  Code release for REPAIR: REnormalizing Permuted Activations for Interpolation Repair
  ☆48Updated last year
• thegregyang / NTK4A
  Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"
  ☆105Updated 4 years ago
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆189Updated 8 months ago
• gpauloski / kfac-pytorch
  Distributed K-FAC preconditioner for PyTorch
  ☆89Updated this week
• libffcv / ffcv-imagenet
  Train ImageNet *fast* in 500 lines of code with FFCV
  ☆144Updated last year
• bobby-he / Neural_Tangent_Kernel
  ☆67Updated 6 years ago
• alecwangcq / KFAC-Pytorch
  Pytorch implementation of KFAC and E-KFAC (Natural Gradient).
  ☆132Updated 6 years ago
• locuslab / edge-of-stability
  ☆70Updated 8 months ago
• tml-epfl / understanding-sam
  Towards Understanding Sharpness-Aware Minimization [ICML 2022]
  ☆35Updated 3 years ago
• ayulockin / LossLandscape
  Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …
  ☆65Updated 5 years ago
• amirgholami / adahessian
  ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning
  ☆277Updated 2 years ago
• michaelsdr / momentumnet
  Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities
  ☆208Updated last year
• preetum / cifar5m
  CIFAR-5m dataset
  ☆39Updated 4 years ago
• cybertronai / pytorch-sso
  PyTorch-SSO: Scalable Second-Order methods in PyTorch
  ☆147Updated last year
• fwcore / mean-field-theory-deep-learning
  paper lists and information on mean-field theory of deep learning
  ☆78Updated 6 years ago
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆123Updated 4 years ago
• Thrandis / EKFAC-pytorch
  Repository containing Pytorch code for EKFAC and K-FAC perconditioners.
  ☆146Updated 2 years ago
• ansuini / IntrinsicDimDeep
  Code for the intrinsic dimensionality estimate of data representations
  ☆83Updated 5 years ago
• verbiiyo / rigl-torch
  Lightweight torch implementation of rigl, a sparse-to-sparse optimizer.
  ☆57Updated 3 years ago
• yaodongyu / Rethink-BiasVariance-Tradeoff
  Rethinking Bias-Variance Trade-off for Generalization of Neural Networks
  ☆49Updated 4 years ago
• AnanyaKumar / transfer_learning
  Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…
  ☆151Updated 2 years ago
• avirmaux / lipEstimation
  ☆59Updated 2 years ago
• noahgolmant / pytorch-hessian-eigenthings
  Efficient PyTorch Hessian eigendecomposition tools!
  ☆376Updated last year
• wiseodd / last_layer_laplace
  Last-layer Laplace approximation code examples
  ☆82Updated 3 years ago