Tools

We strongly believe in enabling researchers to perform rapid design-space exploration and prototyping of novel microarchitectures and systems. To this end, we have developed and released the following open-source design tools.

Please email Tushar Krishna if you need any information about any of these tools.

AI/ML Distributed Platform Modeling Frameworks

LinkDescriptionLanguagePapers
ChakraDistributed ML workload graph representationPythonArxiv
ASTRA-Sim



Distributed ML Communication SimulatorC++ISPASS 2023

ISPASS 2020
GenZPlatform Simulator for LLM Inference. Try Now PythonArxiv

AI/ML Accelerator Modeling Frameworks

LinkDescriptionLanguagePapers
GAMMA DiGamma Specialized Genetic Algorithms for Mapping and HW resource optimization for DNN Accelerators (works with MAESTRO and Timeloop)Python

ICCAD 2020

DATE 2022

MAESTROAnalytical Model for Deep Learning Dataflows and AcceleratorsC++MICRO 2019
SCALE-simCycle-level Simulator for Systolic Arrays C++ISPASS 2020
STONNECycle-level simulator for flexible Dense and Sparse DNN Accelerators C++IISWC 2021

AI/ML Accelerator RTL

LinkDescription Language Papers
FEATHERReconfigurable Accelerator with Data Layout Reordering SupportVerilogISCA 2024
SIGMASparse and Irregular GEMM Accelerator VerilogHPCA 2020
MAERIDNN Accelerator with Reconfigurable InterconnectsBluespec
(generates Verilog)
ASPLOS 2018
Microswitch-NoCNoC specialized for DNN AcceleratorsBluespec
(generates Verilog)
NOCS 2017

Networks-on-Chip (NoC) for Many-core SoCs

LinkDescriptionLanguagePapers
garnetNetwork-on-Chip and Network-on-Package Model inside the gem5 full system simulatorC++

DAC 2020

ISPASS 2009

OpenSMARTParameterizeable NoC RTL generator Bluespec
(generates Verilog)
ISPASS 2017

The whole is greater than the sum of its parts