ML Runtime & Optimization Engineer for Autonomous Driving

Pony.ai; Fremont (CA); On-site; USD 140,000 - 250,000; Full time

Benefits offered by this job

Health Care Plan

Retirement Plan

Life Insurance

Paid Time Off

Family Leave

Disability Insurance

Free Food & Snacks

Job summary

Pony.ai is seeking a Machine Learning Engineer for the ML Runtime & Optimization team in California. You will develop technologies to accelerate training and inferences of AI models used in autonomous driving systems.

You will collaborate with hardware and software teams to optimize parallel algorithms, influence compute platform architecture, and apply efficient DL techniques across the ML stack, including Torch, CUDA and TensorRT.

Qualifications

• BS/MS/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
• 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

Responsibilities

• 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

Skills

C/C++

Python

Model optimization

Profiling

CUDA

TensorRT

Torch

OpenCL

Communication

Education

BS/MS/Ph.D in CS/EE

Tools

CUDA

OpenCL

OpenACC

Torch

TensorRT

Job description

Pony.ai is seeking a Machine Learning Engineer for the ML Runtime & Optimization team in California. You will develop technologies to accelerate training and inferences of AI models used in autonomous driving systems.

You will collaborate with hardware and software teams to optimize parallel algorithms, influence compute platform architecture, and apply efficient DL techniques across the ML stack, including Torch, CUDA and TensorRT.