This repository contains the official JAX implementation for the paper SAFE: Finding Sparse and Flat Minima to Improve Pruning. Our work introduces SAFE, an algorithm designed to find sparse and flat ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results