☆21Jan 23, 2024Updated 2 years ago
Alternatives and similar repositories for MSAM
Users that are interested in MSAM are comparing it to the libraries listed below
Sorting:
- Code for AAAI 2024 paper: CR-SAM: Curvature Regularized Sharpness-Aware Minimization☆12Nov 29, 2024Updated last year
- ☆17Dec 11, 2022Updated 3 years ago
- ☆34Jan 25, 2024Updated 2 years ago
- A repo based on XiLin Li's PSGD repo that extends some of the experiments.☆14Oct 7, 2024Updated last year
- Sharpness-Aware Minimization Leads to Low-Rank Features [NeurIPS 2023]☆29Sep 22, 2023Updated 2 years ago
- This is unofficial repository for Towards Efficient and Scalable Sharpness-Aware Minimization.☆37Apr 15, 2024Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated 7 months ago
- ☆15Apr 26, 2022Updated 3 years ago
- ☆35Dec 5, 2022Updated 3 years ago
- One-Pixel Shortcut: on the Learning Preference of Deep Neural Networks (ICLR 2023 Spotlight)☆14Sep 28, 2025Updated 5 months ago
- [ICLR 2023] Eva: Practical Second-order Optimization with Kronecker-vectorized Approximation☆12Jul 31, 2023Updated 2 years ago
- Source code for "Scalable Transformer for High Dimensional Multivariate Time Series Forecasting" (Accepted by CIKM-24)☆19Jan 13, 2025Updated last year
- The official repo for CVPR2023 highlight paper "Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization".☆85Jun 20, 2023Updated 2 years ago
- ☆57Feb 13, 2023Updated 3 years ago
- LoRA-One: One-Step Full Gradient Could Suffice for Fine-Tuning Large Language Models, Provably and Efficiently (ICML2025 Oral)☆29Oct 22, 2025Updated 5 months ago
- Example codes in the medium post titled "Optuna meets Weights and Biases."☆24Aug 11, 2022Updated 3 years ago
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…☆36Jul 21, 2025Updated 8 months ago
- ☆23Feb 20, 2026Updated last month
- ☆12Mar 16, 2022Updated 4 years ago
- ☆24Jan 29, 2026Updated last month
- Code and data for the paper "In or Out? Fixing ImageNet Out-of-Distribution Detection Evaluation"☆27Aug 22, 2023Updated 2 years ago
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated last year
- Towards Efficient and Effective Adversarial Training, NeurIPS 2021☆17Feb 15, 2022Updated 4 years ago
- 4-bit Shampoo for Memory-Efficient Network Training (NeurIPS 2024)☆13Feb 13, 2025Updated last year
- ☆48Jan 18, 2024Updated 2 years ago
- Weight-Averaged Sharpness-Aware Minimization (NeurIPS 2022)☆28Jan 13, 2023Updated 3 years ago
- ☆16May 14, 2025Updated 10 months ago
- Optimization algorithm which fits a ResNet to CIFAR-10 5x faster than SGD / Adam (with terrible generalization)☆14Oct 20, 2023Updated 2 years ago
- ☆11Oct 31, 2021Updated 4 years ago
- Code for the API, workload execution, and agents underlying the LLMail-Inject Adpative Prompt Injection Challenge☆21Mar 1, 2026Updated 3 weeks ago
- ☆17Dec 7, 2025Updated 3 months ago
- ☆13Apr 25, 2024Updated last year
- This is the project for IRM methods☆12Sep 13, 2021Updated 4 years ago
- ☆56Feb 24, 2026Updated 3 weeks ago
- KANs and MLPs☆12Jun 7, 2024Updated last year
- ☆14Aug 28, 2019Updated 6 years ago
- Train to 94% on CIFAR-10 in 4.4 seconds on a single A100☆12Dec 30, 2023Updated 2 years ago
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [ICLR 2025]☆28Feb 20, 2026Updated last month
- This is a Pytorch Implementation of the DASP algorithm from the paper "Explaining Deep Neural Networks with a Polynomial Time Algorithm f…☆11Jun 12, 2020Updated 5 years ago