53 citations · 54 across the 3 of their papers we have counts for
4 papers
JaxPruner: A concise library for sparsity research
Joo Hyung Lee, Wonpyo Park, Nicole Mitchell +22
This paper introduces JaxPruner, an open-source JAX-based pruning and sparse training library for machine learning research. JaxPruner aims to accelerate research on sparse neural…
Compiler Support for Sparse Tensor Computations in MLIR
Aart J. C. Bik, Penporn Koanantakool, Tatiana Shpeisman +3
Sparse tensors arise in problems in science, engineering, machine learning, and data analytics. Programs that operate on such tensors can exploit sparsity to reduce storage require…
Composable and Modular Code Generation in MLIR: A Structured and Retargetable Approach to Tensor Compiler Construction
Nicolas Vasilache, Oleksandr Zinenko, Aart J. C. Bik +9
Despite significant investment in software infrastructure, machine learning systems, runtimes and compilers do not compose properly. We propose a new design aiming at providing unp…
A Note on Exhaustive State Space Search for Efficient Code Generation
Aart J. C. Bik
This note explores state space search to find efficient instruction sequences that perform particular data manipulations. Once found, the instruction sequences are hard-wired in th…