Hao Kang
about projects notes
← Index

About

I’m an incoming CS PhD student at Carnegie Mellon University, fortunate to be co-advised by Professor Chenyan Xiong and Professor Tianqi Chen. My research moves in two directions: designing hardware-aligned architectures for training foundation models at scale, and building agent-first machine learning systems that agents can understand, operate, and extend.

I’m a core contributor to PithTrain, a compact and agent-native MoE training system. It delivers production-grade performance, including 4D parallelism, compute-communication overlap, and FP8 training, with minimal indirection and shipped agent skills for recurring tasks, in a codebase compact enough that an agent (or a human) can read it end-to-end.


My CS journey all started with a Python tutorial on YouTube. It was immediately clear after the first few hours of watching that once I learned how to program, I could build anything. The pursuit of that kind of freedom has carried me through my undergrad years at CMU and into the work I do today. Never did I anticipate back then how the power of AI would amplify that momentum.

I believe that technology, when guided by an innovative mind, can make a positive impact on this world. To this end, I want to write the code that makes things happen :-)