Popular repositories Loading

1. SparseTIR SparseTIR Public

   SparseTIR: Sparse Tensor Compiler for Deep Learning

   Python 134 14

2. dtr-prototype dtr-prototype Public

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

   Python 132 18

3. nexus nexus Public

   C++ 82 23

4. relay-aot relay-aot Public

   An experimental ahead of time compiler for Relay.

   Python 50 8

5. tutorial tutorial Public

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

   Jupyter Notebook 24 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 24 6