y0ast / pytorch-snippets
Collection of snippets for PyTorch users
☆26Updated 2 years ago
Related projects: ⓘ
- ☆35Updated 2 years ago
- ☆96Updated 2 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆61Updated 4 years ago
- Supporing code for the paper "Bayesian Model Selection, the Marginal Likelihood, and Generalization".☆34Updated 2 years ago
- Bayesianize: A Bayesian neural network wrapper in pytorch☆81Updated 4 months ago
- CIFAR-5m dataset☆39Updated 3 years ago
- ☆52Updated last month
- Last-layer Laplace approximation code examples☆78Updated 2 years ago
- Codebase for Learning Invariances in Neural Networks☆94Updated last year
- ☆32Updated 11 months ago
- ☆14Updated 4 years ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Updated last year
- Code base for SRSGD.☆28Updated 4 years ago
- ☆13Updated last year
- Code for the CVPR 2021 paper: Understanding Failures of Deep Networks via Robust Feature Extraction☆35Updated 2 years ago
- ☆19Updated 2 years ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆77Updated 2 years ago
- SGD with large step sizes learns sparse features [ICML 2023]☆31Updated last year
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆103Updated 2 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆47Updated 5 years ago
- ☆10Updated 2 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 3 years ago
- Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning☆33Updated 4 years ago
- Limitations of the Empirical Fisher Approximation☆45Updated 4 years ago
- Contains code for the NeurIPS 2020 paper by Pan et al., "Continual Deep Learning by FunctionalRegularisation of Memorable Past"☆43Updated 3 years ago
- Code for "Supermasks in Superposition"☆115Updated 11 months ago
- Gradient Starvation: A Learning Proclivity in Neural Networks☆59Updated 3 years ago
- ☆18Updated 11 months ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆35Updated last year