Machine Learning Engineer, ML Runtime & Optimization job opportunity at Pony.ai.



Date2026-02-06T00:31:43.697Z bot
Pony.ai Machine Learning Engineer, ML Runtime & Optimization
Experience: General
Pattern: Full-time
apply Apply Now
Salary:
Status:

ML Runtime & Optimization

Copy Link Report
degreeGeneral
loacation Fremont, United States Of America
loacation Fremont....United States Of America

Founded in 2016 in Silicon Valley, Pony.ai has quickly become a global leader in autonomous mobility and is a pioneer in extending autonomous mobility technologies and services at a rapidly expanding footprint of sites around the world. Operating Robotaxi, Robotruck and Personally Owned Vehicles (POV) business units, Pony.ai is an industry leader in the commercialization of autonomous driving and is committed to developing the safest autonomous driving capabilities on a global scale. Pony.ai’s leading position has been recognized, with CNBC ranking Pony.ai #10 on its CNBC Disruptor list of the 50 most innovative and disruptive tech companies of 2022. In June 2023, Pony.ai was recognized on the XPRIZE and Bessemer Venture Partners inaugural “XB100” 2023 list of the world’s top 100 private deep tech companies, ranking #12 globally. As of August 2023, Pony.ai has accumulated nearly 21 million miles of autonomous driving globally. Pony.ai went public at NASDAQ in Nov. 2024. Responsibility The ML Infrastructure team at Pony.ai provides a set of tools to support and automate the lifecycle of the AI workflow, including model development, evaluation, optimization, deployment, and monitoring. As a Machine Learning Engineer in ML Runtime & Optimization, you will be developing technologies to accelerate the training and inferences of the AI models in autonomous driving systems. This includes: Identifying key applications for current and future autonomous driving problems and performing in-depth analysis and optimization to ensure the best possible performance on current and next-generation compute architectures. Collaborating closely with diverse groups in Pony.ai including both hardware and software to optimize and craft core parallel algorithms as well as to influence the next-generation compute platform architecture design and software infrastructure. Apply model optimization and efficient deep learning techniques to models and optimized ML operator libraries. Work across the entire ML framework/compiler stack (e.g.Torch, CUDA and TensorRT), and system-efficient deep learning models. BS/MS or Ph.D in computer science, electrical engineering or a related discipline. Strong programming skills in C/C++ or Python. Experience on model optimization, quantization or other efficient deep learning techniques Good understanding of hardware performance, regarding CPU or GPU execution model, threads, registers, cache, cost/performance trade-off, etc. Experience with profiling, benchmarking and validating performance for complex computing architectures. Experience in optimizing the utilization of compute resources, identifying and resolving compute and data flow bottlenecks. Strong communication skills and ability to work cross-functionally between software and hardware teams Preferred Qualifications: One or more of the following fields are preferred Experience with parallel programming, ideally CUDA, OpenCL or OpenACC. Experience in computer vision, machine learning and deep learning. Strong knowledge of software design, programming techniques and algorithms. Good knowledge of common deep learning frameworks and libraries. Deep knowledge on system performance, GPU optimization or ML compiler. Compensation and Benefits Base Salary Range: $140,000 - $250,000 Annually Compensation may vary outside of this range depending on many factors, including the candidate’s qualifications, skills, competencies, experience, and location. Base pay is one part of the Total Compensation and this role may be eligible for bonuses/incentives and restricted stock units. Also, we provide the following benefits to the eligible employees: Health Care Plan (Medical, Dental & Vision) Retirement Plan (Traditional and Roth 401k) Life Insurance (Basic, Voluntary & AD&D) Paid Time Off (Vacation & Public Holidays) Family Leave (Maternity, Paternity) Short Term & Long Term Disability Free Food & Snacks

Other Ai Matches

Machine Learning Engineer, ML Runtime & Optimization Applicants are expected to have a solid experience in handling ML Runtime & Optimization related tasks
Research Intern (Deep Learning), 2026 Spring (Master/PhD) Applicants are expected to have a solid experience in handling 2026 Spring (Master/PhD) related tasks
Software Engineer - Perception Infrastructure Applicants are expected to have a solid experience in handling Job related tasks
Software Engineer, Perception (Robotics) Applicants are expected to have a solid experience in handling Perception (Robotics) related tasks
IT Support Specialist Applicants are expected to have a solid experience in handling Job related tasks
(Senior) Software Engineer, Infrastructure (Kubernetes Platform) Applicants are expected to have a solid experience in handling Infrastructure (Kubernetes Platform) related tasks
Software Engineer - Map Generalist Applicants are expected to have a solid experience in handling Job related tasks
Security Operations Engineer Applicants are expected to have a solid experience in handling Job related tasks
Self-Driving Vehicle Operator (contract) Applicants are expected to have a solid experience in handling Job related tasks
Software Engineer, Infrastructure (Onboard System & Toolchain) Applicants are expected to have a solid experience in handling Infrastructure (Onboard System & Toolchain) related tasks