fattorib / Flax-ResNets
CIFAR10 ResNets implemented in JAX+Flax
☆11Updated 2 years ago
Alternatives and similar repositories for Flax-ResNets:
Users that are interested in Flax-ResNets are comparing it to the libraries listed below
- Latest Weight Averaging (NeurIPS HITY 2022)☆28Updated last year
- ☆21Updated last year
- ☆51Updated 7 months ago
- DiWA: Diverse Weight Averaging for Out-of-Distribution Generalization☆29Updated last year
- ☆21Updated 2 years ago
- Deep Learning & Information Bottleneck☆53Updated last year
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆79Updated last year
- Recycling diverse models☆44Updated 2 years ago
- ☆15Updated last year
- ☆17Updated 2 years ago
- ☆19Updated 2 years ago
- ☆16Updated 6 months ago
- ☆43Updated 2 years ago
- ☆29Updated 2 years ago
- ☆35Updated last year
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- A modern look at the relationship between sharpness and generalization [ICML 2023]☆43Updated last year
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- ImageNet-12k subset of ImageNet-21k (fall11)☆21Updated last year
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆35Updated 2 years ago
- Gradient-based Hyperparameter Optimization Over Long Horizons☆13Updated 3 years ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆16Updated 2 years ago
- ☆27Updated 6 months ago
- ☆57Updated last year
- ☆11Updated 2 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆62Updated 4 years ago
- ☆28Updated last year
- Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)☆20Updated 2 months ago
- Collection of snippets for PyTorch users☆26Updated 2 years ago