A tiny Research Scientist and Engineering Manager
at ByteDance
May 2023-Present
Senior Staff-level Research Scientist and Engineering Manager
Worked at GenAI, distributed training, AI compiler, AI intrastructure, PyTorch, OpenAI Triton, Tensorflow, ONNX, MLIR, OpenXLA, NPU ASICs
Delivered 1,000s models for in-house ASICs.
Managed a team and built veScale, a PyTorch native LLM training framework.
Mentored an intern and built ShadowKV, high-throughput long-context LLM inference.
Led a team and built Flux, a fast communication-overlapping library for tensor parallelism on GPUs.
Managed and led ByteIR team.
Managed ByteMLPerf team.
Bootstrapped some forward-looking projects, across LLM and vision, including many aspects of MLSys, compiler, automation, and acceleration.
ByteDance Seed/AML
Jun 2021-May 2023
Staff-level Research Scientist
Worked at AI compiler, AI intrastructure, libraries, PyTorch, MLIR, XLA, NPU ASICs
Built a compiler for in-house ASICs.
Led a team and built ByteIR, a model compilation solution for various hardware.
Led a team and built ByteMLPerf, an AI accelerator benchmarking tool.
ByteDance AML
Sept 2018-Jun 2021
Senior Software Engineer
Worked at AI
intrastructure, libraries, PyTorch, ONNX, NPUs
Microsoft Cloud & AI
July 2017-Aug 2018
Software Engineer
Worked at AI intrastructure,
libraries and ONNX
Microsoft AI & Research
Fall 2009-Summer 2017
Research Assistant
Researched in high-performance
computing and compiler optimizations
IMPACT Lab., UIUC
Summer 2012
Intern
Developed real-time image inpainting for
Tegra 3
NVIDIA
Feb. 2008-July 2009
Research Assistant and Engineer
Developed a novel
high-frequency (50MHz) real-time ultrasonic imaging system, using FPGAs and GPGPU. The
system became a commercial product of a startup.
Ultrasonic Imaging Lab., NTU
Jul. 2017
Ph.D. in Electrical and Computer Engineering (ECE)
Under Dr. Wen-mei Hwu's supervision.
University of Illinois at Urbana-Champaign (UIUC), IL
Aug. 2014
Master in ECE
Under Dr. Wen-mei Hwu's
supervision.
University of Illinois at Urbana-Champaign (UIUC), IL
Jun. 2007
BS in Electrical Engineering (EE), with a Minor degree in Mathematics
National Taiwan University (NTU), Taipei, Taiwan
Aug. 2006-May 2007
Visiting student in ECE
UIUC
2012-2013
Dan Vivoli Endowed Fellowship
ECE, UIUC
2009-2011
Integrative Graduate Education and Research Traineeship (IGERT): Neuroengineering
NSF, USA
2006-2007
Taiwan Merit Scholarship
Taiwan
2005
Undergraduate Student Research Fellowship
National Science Council, Taiwan
2005
Pan Wen-Yuan Scholarship
Pan Wen-Yuan Foundation, Taiwan
3 times
Presidential Awards
NTU, Taiwan
2001
Gold medal in Asian Pacific Mathematics Olympiad (APMO)