pnnl / torchntk

Related projects ⓘ

Alternatives and complementary repositories for torchntk

• tml-epfl / understanding-sam
  Towards Understanding Sharpness-Aware Minimization [ICML 2022]
  ☆35Updated 2 years ago
• gortizji / linearized-networks
  Source code of "What can linearized neural networks actually say about generalization?
  ☆18Updated 3 years ago
• fKunstner / noise-sgd-adam-sign
  ☆15Updated last year
• locuslab / orthogonal-convolutions
  Implementations of orthogonal and semi-orthogonal convolutions in the Fourier domain with applications to adversarial robustness
  ☆42Updated 3 years ago
• izmailovpavel / bnn_covariate_shift
  Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"
  ☆33Updated 2 years ago
• tml-epfl / sharpness-vs-generalization
  A modern look at the relationship between sharpness and generalization [ICML 2023]
  ☆43Updated last year
• bobby-he / Neural_Tangent_Kernel
  ☆67Updated 5 years ago
• JonasGeiping / fullbatchtraining
  Training vision models with full-batch gradient descent and regularization
  ☆38Updated last year
• hushon / JAX-ResNet-CIFAR10
  Simple CIFAR10 ResNet example with JAX.
  ☆21Updated 3 years ago
• RuntianZ / doro
  Distributional and Outlier Robust Optimization (ICML 2021)
  ☆27Updated 3 years ago
• rahimentezari / PermutationInvariance
  ☆23Updated 2 years ago
• facebookresearch / BalancingGroups
  Simple data balancing baselines for worst-group-accuracy benchmarks.
  ☆40Updated last year
• team-approx-bayes / fromp
  Contains code for the NeurIPS 2020 paper by Pan et al., "Continual Deep Learning by FunctionalRegularisation of Memorable Past"
  ☆44Updated 4 years ago
• gibipara92 / learning-explanations-hard-to-vary
  Code to implement the AND-mask and geometric mean to do gradient based optimization, from the paper "Learning explanations that are hard …
  ☆39Updated 4 years ago
• tml-epfl / sgd-sparse-features
  SGD with large step sizes learns sparse features [ICML 2023]
  ☆32Updated last year
• bobby-he / bayesian-ntk
  Code to accompany paper 'Bayesian Deep Ensembles via the Neural Tangent Kernel'
  ☆27Updated 3 years ago
• ppope / dimensions
  Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J
  ☆65Updated 7 months ago
• team-approx-bayes / kpriors
  Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.
  ☆16Updated 2 years ago
• MadryLab / BREEDS-Benchmarks
  ☆55Updated 4 years ago
• deep-lab / DeepnetHessian
  Measurements of Three-Level Hierarchical Structure in the Outliers in the Spectrum of Deepnet Hessians (ICML 2019)
  ☆17Updated 5 years ago
• nitarshan / robust-generalization-measures
  Official code for "In Search of Robust Measures of Generalization" (NeurIPS 2020)
  ☆28Updated 3 years ago
• avirmaux / lipEstimation
  ☆57Updated last year
• YannDubs / Invariant-Self-Supervised-Learning
  Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"
  ☆40Updated last year
• thudzj / ELLA
  Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')
  ☆16Updated 2 years ago
• samuelstanton / gnosis
  Code to reproduce experiments from 'Does Knowledge Distillation Really Work' a paper which appeared in the 2021 NeurIPS proceedings.
  ☆33Updated last year
• xu-ji / information-bottleneck
  Deep Learning & Information Bottleneck
  ☆50Updated last year
• team-approx-bayes / bayesian-sam
  Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.
  ☆23Updated last year
• iamalexkorotin / Wasserstein2Barycenters
  PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)
  ☆23Updated 2 years ago
• yaodongyu / Rethink-BiasVariance-Tradeoff
  Rethinking Bias-Variance Trade-off for Generalization of Neural Networks
  ☆49Updated 3 years ago