Selected Projects

  • 2021 - 2022
    SparseTIR, A Tensor-Level Abstraction for Sparse Operator Optimization in Deep Learning
  • 2020 Fall - Now
    Apache TVM, An End-to-End Machine Learning Compiler Framework
  • 2020 Spring
    • A toy compiler implemented in Java, from Mx* (a C- and Java-like language) to RISC-V assembly code.
    • Implemented many effective optimizations. The generated code has performance close to GCC O2.
    • More than 15k+ lines of code overall.
  • 2019 Summer
    Distributed Hash Table
    • Implemented two DHT protocols, Chord and Kademlia, in Go Language.
    • Implemented an instant chat room system based on the Chord protocol.


  • 2022 Summer
    Machine Learning Compilation
    • Teaching assistant of this online course.
    • Prepare and release course assignments. Answer questions in the discussion page.
  • 2020 Summer
    Principle and Practice of Computer Algorithms
    • Leader teaching assistant of this SJTU course.
    • Advised students to implement a RISC-V simulator.
    • Advised students to implement two Distributed Hash Table protocols, Chord and Kademlia, in Go Language.
  • 2020 Spring
    Data Structure (Honor)
    • Leader teaching assistant of this SJTU course.
    • Taught advanced data structures which students usually do not learn in class.
    • Prepared the course assignments, projects and programming exams.

Honors and Awards

  • 2022
    • Shanghai Excellent Graduate Award
  • 2020
    • National Scholarship (Top 0.2% nationwide)
  • 2018
    • The 4th place of the 2018 ICPC Asia Singapore Regional Contests