Popular repositories Loading

1. SparseTIR SparseTIR Public

   SparseTIR: Sparse Tensor Compiler for Deep Learning

   Python 131 14

2. dtr-prototype dtr-prototype Public

   Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616

   Python 129 16

3. nexus nexus Public

   C++ 82 22

4. relay-aot relay-aot Public

   An experimental ahead of time compiler for Relay.

   Python 51 8

5. tutorial tutorial Public

   A self-contained version of the tutorial which can be easily cloned and viewed by others.

   Jupyter Notebook 25 14

6. sparsetir-artifact sparsetir-artifact Public

   Repository for artifact evaluation of ASPLOS 2023 paper "SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learning"

   Python 23 6