fattorib / Flax-ResNets
CIFAR10 ResNets implemented in JAX+Flax
☆12Updated 3 years ago
Alternatives and similar repositories for Flax-ResNets:
Users that are interested in Flax-ResNets are comparing it to the libraries listed below
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆36Updated 2 years ago
- Recycling diverse models☆44Updated 2 years ago
- ☆18Updated 9 months ago
- ☆16Updated last year
- ☆52Updated 6 months ago
- ☆10Updated 2 years ago
- Latest Weight Averaging (NeurIPS HITY 2022)☆30Updated last year
- Blog post☆17Updated last year
- DiWA: Diverse Weight Averaging for Out-of-Distribution Generalization☆29Updated 2 years ago
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆80Updated last year
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆65Updated 6 months ago
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- ☆35Updated last year
- ☆22Updated 2 years ago
- ☆29Updated 2 years ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- ☆37Updated 3 years ago
- ☆21Updated 2 years ago
- A modern look at the relationship between sharpness and generalization [ICML 2023]☆43Updated last year
- Source code of "What can linearized neural networks actually say about generalization?☆20Updated 3 years ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆35Updated 2 years ago
- Codes for the paper The emergence of clusters in self-attention dynamics.☆15Updated last year
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆65Updated 4 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆62Updated 3 years ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- ☆19Updated 3 years ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆71Updated 5 months ago
- Gradient-based Hyperparameter Optimization Over Long Horizons☆13Updated 3 years ago
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆58Updated last year