index.html

---
layout: home
author_profile: true
---

<div style="padding-left: 10%;">
  <h1> About Me </h1>
  <p> I am Mengdi Wu (吴梦迪), a second year PhD in the <a href="https://csd.cmu.edu/">Computer Science Department</a> at Carnegie Mellon University, advised by <a href="https://www.cs.cmu.edu/~zhihaoj2/">Zhihao Jia</a>. I graduated from <a href="https://iiis.tsinghua.edu.cn/en/yaoclass/">Yao Class</a> at Tsinghua University in 2023. I am interested in machine learning systems.</p>  
  
  <h2> Research Experience </h2>
  <p style="font-size: 16px"> I've learned a lot of exciting knowledege and obtained useful skills from every research experience, especially from my great collaborators. </p>
  
  
  <table>
      <th>2022.02~2022.08</th> <td> I visited CMU as a research intern, where I worked with <a href="https://www.cs.cmu.edu/~zhihaoj2/">Zhihao Jia</a> on accelerating distributed DL training. </td>
    </tr>
    <tr>
      <th>2020.11~2021.11</th> <td> I joined <a href="https://group.iiis.tsinghua.edu.cn/~maks/">ARChip Lab</a> led by <a href="https://group.iiis.tsinghua.edu.cn/~maks/leader.html">Kaisheng Ma</a> at Tsinghua University, and worked on designing architectures and schedulers for DNN accelerators. </td>
    </tr>
  </table>
  
  <h2> Awards </h2>
  <div class="award__">
    <ul>
      <li>
        <a href="https://compiler.educg.net/2020CSCC">National Student Computer System Capability Challenge 2020 (Compiler track)</a>, 2nd Price [<a href="https://github.com/Cutepiler/Cutepiler-Sysy2020">Cutepiler</a>]
        <p class="comment__">With Jiatu Li, Runda Liu, and Zhidong Wang. </p>
      </li>
      <li>
        National Olympiad in Informatics, China, Gold Medal
      </li>
    </ul>
  </div>

  <h2> Publications </h2>
  <div class="project__">
    <ul>
      <li class="project__title">
        <a href="https://arxiv.org/abs/2406.17145">
          GraphPipe: Improving the Performance and Scalability of DNN Training with Graph Pipeline Parallelism
        </a>
      </li>
      <li class="project__author">
        Byungsoo Jeon<sup>*</sup>, Mengdi Wu<sup>*</sup>, Shiyi Cao<sup>*</sup>, Sunghyun Kim<sup>*</sup>, Sunghyun Park, Neeraj Aggarwal, Colin Unger, Daiyaan Arfeen, Peiyuan Liao, Xupeng Miao, Mohammad Alizadeh, Gregory R. Ganger, Tianqi Chen, Zhihao Jia
      </li>
      <li class="project__publish">
        ASPLOS 2025
      </li>
    </ul>
  </div>
  <div class="project__">
    <ul>
      <li class="project__title">
        <a>
          LACO: A Latency-Constraint Offline Neural Network Scheduler towards Reliable Self-Driving Perception
        </a>
      </li>
      <li class="project__author">
        Zhanhong Tan, Zijian Zhu, Mengdi Wu, Kaisheng Ma
      </li>
      <li class="project__publish">
        ICCAD 2024
      </li>
    </ul>
  </div>
  <div class="project__">
    <ul>
      <li class="project__title">
        <a href="https://proceedings.mlr.press/v162/dong22a.html">
          Finding the Task-Optimal Low-Bit Sub-Distribution in Deep Neural Networks
        </a>
      </li>
      <li class="project__author">
        Runpei Dong<sup>*</sup>, Zhanhong Tan<sup>*</sup>, Mengdi Wu, Linfeng Zhang, Kaisheng Ma
      </li>
      <li class="project__publish">
        ICML 2022
      </li>
    </ul>
  </div>
<!--   
  <h2> Miscellaneous </h2>
  <div class="award__">
    <ul>
      <li>
        I am a beginner in Japanese.
      </li>
      <li>
        I would also like to advertize my excellent friends in Tsinghua. I have learnt a lot on study, research and life from them:
          <a href="https://zihanhu.cn/">Zihan Hu</a>,
          <a href="https://ziqing-huang.github.io/">Ziqing Huang</a>,
          <a href="https://tu-yaxin.github.io/">Yaxin Tu</a>,
          and my boyfriend <a href="http://ljt12138.github.io/">Jiatu Li</a>.
      </li>
    </ul>
  </div>
   -->

</div>