Congjie He is a PhD student in the School of Informatics at the University of Edinburgh. His research focuses on building AI training and inference infrastructure in large-scale distributed memory architectures, such as the Cerebras WSE-3 and GPU Super-Pods, and involves the co-design of models, systems, and hardware. He is also the leading author of WaferLLM, and his research has been published in conferences such as OSDI, NeurIPS, and ICML.