SAMPL
SparseTIR: Sparse Tensor Compiler for Deep Learning
Python 143 14
Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
Python 133 18
An experimental ahead of time compiler for Relay.
Python 49 8
Repository for artifact evaluation of ASPLOS 2023 paper "SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learning"
Python 25 7
A self-contained version of the tutorial which can be easily cloned and viewed by others.
Jupyter Notebook 24 12