Cheng Wan

United States

Join Prog.AI to see contacts

Summary

🤩

Rockstar

🎓

Top School

Cheng Wan is a Member of Technical Staff and Ph.D. student in Computer Science at Georgia Tech with five years of professional experience building machine learning systems and inference infrastructure at organizations including xAI, ByteDance, AWS, and Microsoft Research. He brings deep hands-on expertise in ML system engineering and distributed computation, having contributed to production inference stacks and applied research during multiple internships. Cheng is also an active backend contributor to the popular Deep Graph Library (DGL), where he fixed OpenMP/PyTorch memory issues and improved distributed partitioning and METIS-based partition strategies—work that reflects a strong command of both Python and C++ internals. His background spans academia and industry, combining rigorous research training (Rice University and Shanghai Jiao Tong University) with practical delivery in large-scale ML teams. Notably, his contributions often focus on reliability and performance optimizations that are easy to miss but materially improve system robustness. He is based in the United States and specializes in turning complex graph and ML research problems into production-ready solutions.

5 years of coding experience

1 year of employment as a software developer

Graduate Student, Computer Science, Graduate Student, Computer Science at Georgia Institute of Technology

Graduate Student, Electrical and Computer Engineering, Graduate Student, Electrical and Computer Engineering at Rice University

Bachelor of Engineering - BE, Computer Science, Bachelor of Engineering - BE, Computer Science at Shanghai Jiao Tong University

Stackoverflow

Stats

1reputation

0reached

0answers

0questions

Github Skills (8)

deeplearning-ai10

c-language10

deep-learning10

cprogramming-language10

python10

distributed-computing10

pytorch9

openmp8

Programming languages (1)

Python

Github contributions (5)

dmlc/dgl

Aug 2020 - Mar 2022

Python package built to ease deep learning on graph, on top of existing DL frameworks.

Role in this project:

Back-end Developer

Contributions:4 reviews, 9 commits, 6 PRs in 1 year 7 months

Contributions summary:Cheng primarily contributed to bug fixes and improvements related to the DGL (Deep Graph Library) Python package. Their work involved resolving OpenMP compatibility issues, addressing typos, and enhancing the distributed partition functionality. They also contributed to memory leak prevention within the PyTorch backend and incorporated METIS partitioning with communication volume minimization. The contributions span across various Python modules and C++ code, demonstrating a strong understanding of the library's internal workings.

pytorchpythondeep-learningmachine-learninggraph-neural-networks

GATECH-EIC/BNS-GCN

Feb 2022 - Nov 2022

Contributions:25 commits, 17 pushes, 2 branches in 9 months

Find and Hire Top DevelopersWe’ve analyzed the programming source code of over 60 million software developers on GitHub and scored them by 50,000 skills. Sign-up on Prog,AI to search for software developers.

Request Free Trial