Jiageng Wu is a product leader and former software engineer with five years of experience bridging ML engineering, cloud services, and product strategy across Beijing and global teams. Currently a Senior Product Manager at Beijing Frontis, he has driven product efforts after roles at ByteDance, EY, Xiaomi, and hands-on engineering at Lenovo, combining technical execution with strategic insight. Academically grounded with degrees from Peking University, Brandeis, and doctoral study in computational mathematics, he blends rigorous research training with practical product delivery. An active contributor to the prominent Torch-MLIR project, he implemented complex PyTorch-to-MLIR conversion patterns—evidence of deep systems and ML compiler skills that inform his product decisions. Known for translating low-level technical complexity into actionable roadmaps, he seeks roles where applied ML, cloud infrastructure, and product intersect.
5 years of coding experience
2 years of employment as a software developer
Bachelor of Engineering - BE, Internet of Things, Bachelor of Engineering - BE, Internet of Things at 北京邮电大学
Doctorate Degree, Computational Mathematics, Doctorate Degree, Computational Mathematics at 吉林大学
Master's degree, Computer Science, 3.5, Master's degree, Computer Science, 3.5 at Brandeis University
The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
Role in this project:
ML Engineer
Contributions:129 reviews, 18 commits, 89 PRs in 2 months
Contributions summary:Jiageng contributed significantly to the Torch-MLIR project, primarily focusing on the conversion of PyTorch operations to the MLIR ecosystem. Their work involved implementing and refining conversion patterns for various PyTorch operations, including those related to MHLO and CHLO dialects. The user's contributions included supporting dynamic shapes, implementing reduce-like and pooling-like op conversions, and adding support for new ops such as `aten.cat` and `aten.clamp`.
The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
Contributions:203 pushes, 62 branches in 2 years 3 months
pytorchmlirtorchecosystemwandb
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.