Staff Machine Learning Engineer

XPENG; Santa Clara (CA); On-site; USD 215,280 - 364,320; Full time

Benefits offered by this job

Snacks, lunches, dinners, and fun activities

Job summary

XPENG is seeking a Machine Learning Engineer to enhance Traffic Sign Recognition for autonomous driving systems. This role involves the full lifecycle of model development, from data preparation to deployment.

Candidates should bring strong experience in computer vision and object detection, with the ability to optimize models for production environments. The compensation package is competitive, and the work has real-world impact on vehicle safety.

Qualifications

• 3-5 years of experience with computer vision models.
• Strong hands-on experience in object detection.
• Proficiency in Python and deep learning frameworks.

Responsibilities

• Develop and improve 2D traffic sign detection models.
• Analyze scenarios and failure cases.
• Prepare training and evaluation datasets.

Skills

Computer vision

Deep learning

Data analysis

Object detection

Education

Master’s or PhD in Computer Science or related field

Tools

PyTorch

TensorFlow

Job description

XPENG is a leading smart technology company at the forefront of innovation, integrating advanced AI and autonomous driving technologies into its vehicles, including electric vehicles (EVs), electric vertical take‑off and landing (eVTOL) aircraft, and robotics. With a strong focus on intelligent mobility, XPENG is dedicated to reshaping the future of transportation through cutting‑edge R&D in AI, machine learning, and smart connectivity.

About the Role

We are looking for a strong Machine Learning Engineer / Computer Vision Engineer to work on Traffic Sign Recognition (TSR) 2D detection for production autonomous driving systems. In this role, you will be responsible for the full lifecycle of TSR model development, including scenario analysis, data preparation, model training, evaluation, optimization, quantization, and deployment. You will work closely with perception, data, infrastructure, and deployment teams to improve traffic sign detection performance across diverse real‑world driving scenarios. This role is ideal for candidates who enjoy solving practical computer vision problems, building reliable model iteration pipelines, and bringing perception models from offline training to onboard production systems.

What You’ll Do

Develop and improve 2D traffic sign detection models for autonomous driving perception systems.

Analyze TSR‑related scenarios and failure cases, including missed detections, false positives, occlusions, small objects, rare signs, region‑specific signs, and adverse weather or lighting conditions.

Prepare, clean, curate, and analyze training and evaluation datasets for TSR model iteration.

Design and execute model training experiments, including data sampling, augmentation, loss tuning, class imbalance handling, and hard‑case mining.

Build and maintain evaluation pipelines for TSR models, including offline metrics, scenario‑based evaluation, regression testing, and error analysis.

Collaborate with data teams to define mining strategies for long‑tail TSR scenarios and improve dataset coverage.

Optimize models for production deployment, including ONNX / TensorRT / quantization / inference acceleration.

Work with deployment and platform teams to validate model performance on onboard or edge compute platforms.

Track model performance across versions and support continuous improvement through data‑model‑evaluation feedback loops.

Debug issues across the full stack, including data quality, labeling, model behavior, evaluation mismatch, and deployment consistency.

Basic Qualifications

• Master’s, or PhD degree in Computer Science, Electrical Engineering, Robotics, Computer Vision, Machine Learning, or a related field.
• 3‑5 years of strong hands‑on experience with computer vision models, especially object detection.
• Experience with detection architectures such as YOLO, Faster R‑CNN, DETR/Deformable DETR, RT‑DETR, RTMDet, or similar models.
• Proficiency in Python and deep learning frameworks such as PyTorch or TensorFlow.
• Solid understanding of object detection training workflows, including dataset preparation, augmentation, loss functions, evaluation metrics, and model debugging.
• Experience with common detection metrics such as mAP, precision/recall, false positive/false negative analysis, and class‑level performance breakdown.
• Strong data analysis and problem‑solving skills.
• Ability to work cross‑functionally with model, data, infrastructure, and deployment teams.

Preferred Qualifications

• Experience in autonomous driving, ADAS, robotics, or safety‑critical perception systems.
• Experience with traffic sign recognition, traffic light recognition, road object detection, or small‑object detection.
• Familiarity with long‑tail scenario mining, hard negative mining, class imbalance handling, and dataset curation.
• Experience with ONNX, TensorRT, model quantization, C++ inference pipelines, CUDA, or edge deployment.
• Experience debugging training‑to‑deployment consistency issues, including preprocessing mismatch, postprocessing mismatch, quantization accuracy drop, or runtime performance bottlenecks.
• Familiarity with large‑scale data pipelines, scenario tagging, or automated data mining workflows.
• Strong engineering discipline in experiment tracking, reproducibility, regression testing, and model version management.

What Success Looks Like

• Improve TSR detection performance across both common and long‑tail traffic sign scenarios.
• Build reliable data and evaluation workflows to support fast model iteration.
• Identify and prioritize high‑impact failure modes through scenario analysis and data mining.
• Deliver deployable TSR models with strong accuracy, latency, and robust tradeoffs.
• Help establish a scalable data‑model‑evaluation‑deployment loop for production TSR development.

Why Join Us

Work on production of autonomous driving perception systems with real‑world impact.

Own an important perception task that directly affects driving safety, rule understanding, and product quality.

Collaborate with strong teams across model development, data, deployment, and vehicle platforms.

Gain hands‑on experience across the full model lifecycle: from data and training to evaluation, optimization, quantization, and onboard deployment.

Competitive compensation package.

Snacks, lunches, dinners, and fun activities.

Compensation

The base salary range for this full‑time position is $215,280 – $364,320, in addition to bonus, equity, and benefits. Our salary ranges are determined by role, level, and location. The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position across all U.S. locations. Within the range, individual pay is determined by work location and additional factors, including job‑related skills, experience, and relevant education or training.

Equal Employment Opportunity Statement

We are an Equal Opportunity Employer. It is our policy to provide equal employment opportunities to all qualified persons without regard to race, age, color, sex, sexual orientation, religion, national origin, disability, veteran status or marital status or any other prescribed category set forth in federal or state regulations.