Yu-te Cheng is a Staff Engineer based in California with 11 years of experience building production-grade deep learning systems for autonomous driving. Currently at XPENG, he leads end-to-end AI model development across 3D/2D BEV, vision-language models, and planning stacks, bridging perception and control. Previously at NVIDIA he shipped deep learning software for autonomous driving and contributed to the pytorch/tensorrt project by adding key PyTorch ops (e.g., aten::transpose, aten::layer_norm) to accelerate models on NVIDIA GPUs. Trained at Carnegie Mellon in robotics and computer vision, he combines research rigor with production engineering, often focusing on model-compiler and deployment robustness that quietly reduce failure modes in real-world driving stacks.
11 years of coding experience
5 years of employment as a software developer
Master's degree Robotics & Computer Vision, Master's degree Robotics & Computer Vision at Carnegie Mellon University
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Role in this project:
ML Engineer
Contributions:7 commits, 13 PRs, 12 pushes in 4 months
Contributions summary:Yu-te primarily contributed to the `pytorch/tensorrt` repository by implementing and testing support for PyTorch operations within the TensorRT environment. This included adding the `aten::transpose` and `aten::layer_norm` operations, enabling more PyTorch models to run efficiently on NVIDIA GPUs. Furthermore, the user addressed linter errors and made general code improvements to ensure code quality and maintainability.
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT
Contributions:12 commits in 3 months
cudapytorchnvidiagpucompiler
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.