Open-source Tools and Contributions

Open-source Development

GAMMA: An autonomous mapping search tool for DNN accelerators

[Source code]

GAMMA-Timeloop: A GAMMA extension for Timeloop (a popular DNN analyitical model from NVIDIA research)

[Source code] [Timeloop Project Page]

Frame: Fast Roofline Analytical Modeling and Estimation

[Source code]

JAX-Pruning: A JAX implementation of structure and unstructure pruning

[Source code]

Open-source Contribution

MAESTRO: A cost and benefit model of dataflows in deep neural network accelerators

[Project page][Source code]

Code Release for Published Papers

Implementation of ConfuciuX: Autonomous Hardware Resource Assignment for DNN Accelerators using Reinforcement Learning

[Source code] [Paper]