Department of Computer Science, Cornell University; Google; Google DeepMind
Yucheng Lu, Shivani Agrawal, Suvinay Subramanian, Oleg Rybakov, Christopher De Sa, Amir Yazdanbakhsh
STEP: Learning N:M Structured Sparsity Masks from Scratch with Precondition