Popular repositories

1. dtr-prototype Public

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

   Python 121 15

2. SparseTIR Public

   SparseTIR: Sparse Tensor Compiler for Deep Learning

   Python 109 10

3. nexus Public

   C++ 71 22

4. relay-aot Public

   An experimental ahead of time compiler for Relay.

   Python 51 8

5. tutorial Public

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

   Jupyter Notebook 24 13

6. sparsetir-artifact Public

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

   Python 15 2