/mase

Machine-Learning Accelerator System Exploration Tools

Primary LanguagePythonOtherNOASSERTION

Machine-Learning Accelerator System Exploration Tools

Contributors Forks Stargazers Issues Doc

Overview

Machine learning accelerators have been used extensively to compute models with high performance and low power. Unfortunately, the development pace of ML models is much faster than the accelerator design cycle, leading to frequent changes in the hardware architecture requirements, rendering many accelerators obsolete. Existing design tools and frameworks can provide quick accelerator prototyping, but only for a limited range of models that fit into a single hardware device. With the emergence of large language models such as GPT-3, there is an increased need for hardware prototyping of large models within a many-accelerator system to ensure the hardware can scale with ever-growing model sizes.

MASE provides an efficient and scalable approach for exploring accelerator systems to compute large ML models by directly mapping onto an efficient streaming accelerator system. Over a set of ML models, MASE can achieve better energy efficiency to GPUs when computing inference for recent transformer models.

Alt text

MASE Publications

  • Fast Prototyping Next-Generation Accelerators for New ML Models using MASE: ML Accelerator System Exploration, link
    @article{cheng2023fast,
    title={Fast prototyping next-generation accelerators for new ml models using mase: Ml accelerator system exploration},
    author={Cheng, Jianyi and Zhang, Cheng and Yu, Zhewen and Montgomerie-Corcoran, Alex and Xiao, Can and Bouganis, Christos-Savvas and Zhao, Yiren},
    journal={arXiv preprint arXiv:2307.15517},
    year={2023}}
    
  • MASE: An Efficient Representation for Software-Defined ML Hardware System Exploration, link
    @article{zhangmase,
    title={MASE: An Efficient Representation for Software-Defined ML Hardware System Exploration},
    author={Zhang, Cheng and Cheng, Jianyi and Yu, Zhewen and Zhao, Yiren}}
    

Repository structure

This repo contains the following directories:

  • components - Internal hardware library
  • scripts - Installation scripts
  • machop - MASE's software stack
  • hls - HLS component of MASE
  • mlir-air - MLIR AIR for ACAP devices
  • docs - Documentation
  • Docker - Docker container configurations

MASE Dev Meetings

Donation

If you think MASE is helpful, please donate for our work, we appreciate your support!