LOG-postech / SasshaLinks
[ICML 2025] Official Pytorch code for "SASSHA: Sharpness-aware Adaptive Second-order Optimization With Stable Hessian Approximation"
β20Updated 2 months ago
Alternatives and similar repositories for Sassha
Users that are interested in Sassha are comparing it to the libraries listed below
Sorting:
- [UAI 2025] Official code for reproducing paper "Critical Influence of Overparameterization on Sharpness-aware Minimization"β19Updated 5 months ago
- π¨ Malet (Machine Learning Experiment Tool) is a tool for efficient machine learning experiment execution, logging, analysis, and plot maβ¦β17Updated 6 months ago
- β17Updated 7 months ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networksβ760Updated 3 months ago
- β25Updated last year
- β28Updated 8 months ago
- β17Updated 2 years ago
- PyTorch implementation of Mixer-nano (#parameters is 0.67M, originally Mixer-S/16 has 18M) with 90.83 % acc. on CIFAR-10. Training from sβ¦β35Updated 3 years ago
- [NeurIPS 2021] code for "Taxonomizing local versus global structure in neural network loss landscapes" https://arxiv.org/abs/2107.11228β19Updated 3 years ago
- Efficient PyTorch Hessian eigendecomposition tools!β379Updated last year
- Pretrained models on CIFAR10/100 in PyTorchβ370Updated 5 months ago
- A curated list of papers of interesting empirical study and insight on deep learning. Continually updating...β376Updated last week
- This repository contains a Pytorch implementation of the paper "The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks"β¦β334Updated 2 years ago
- PyTorch implementation for Vision Transformer[Dosovitskiy, A.(ICLR'21)] modified to obtain over 90% accuracy FROM SCRATCH on CIFAR-10 witβ¦β197Updated last year
- β193Updated 4 years ago
- β71Updated 10 months ago
- Comparison of method "Pruning at initialization prior to training" (Synflow/SNIP/GraSP) in PyTorchβ17Updated last year
- Sharpness-Aware Minimization Leads to Low-Rank Features [NeurIPS 2023]β28Updated 2 years ago
- Pretrained TorchVision models on CIFAR10 dataset (with weights)β682Updated 2 years ago
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.β111Updated 6 years ago
- β36Updated 2 years ago
- β605Updated 2 months ago
- tiny-imagenet dataset downloader & reader using tensorflow_datasets (tfds) apiβ20Updated 6 years ago
- Code reproducing Neural Collapse phenomenon on MSE and cross-entropy lossβ14Updated 3 years ago
- A repository in preparation for open-sourcing lottery ticket hypothesis code.β632Updated 3 years ago
- Lightweight torch implementation of rigl, a sparse-to-sparse optimizer.β59Updated 3 years ago
- Approximating neural network loss landscapes in low-dimensional parameter subspaces for PyTorchβ343Updated last year
- Source code of "Task arithmetic in the tangent space: Improved editing of pre-trained models".β105Updated 2 years ago
- β227Updated last year
- PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.β431Updated 2 years ago