Machine Learning Engineer, ML Runtime & Optimization at pony.ai

Fremont, California, United States

Apply Now

Not SpecifiedCompensation

Mid-level (3 to 4 years), Senior (5 to 8 years)Experience Level

Full TimeJob Type

UnknownVisa

Autonomous DrivingIndustries

Requirements

BS/MS or Ph.D in computer science, electrical engineering, or a related discipline
Strong programming skills in C/C++ or Python
Experience with model optimization, quantization, or other efficient deep learning techniques
Good understanding of hardware performance, including CPU/GPU execution models, threads, registers, cache, and cost/performance trade-offs
Experience with profiling, benchmarking, and validating performance for complex computing architectures
Experience in optimizing compute resource utilization and resolving bottlenecks
Experience with parallel programming

Responsibilities

Develop technologies to accelerate the training and inference of AI models for autonomous driving systems
Identify key applications and perform in-depth analysis and optimization for autonomous driving problems
Collaborate with hardware and software teams to optimize algorithms and influence compute platform design
Apply model optimization and efficient deep learning techniques
Work across the ML framework/compiler stack (e.g., Torch, CUDA, TensorRT) and create system-efficient deep learning models

Skills

C/C++

Python

PyTorch

CUDA

TensorRT

model optimization

quantization

profiling

benchmarking

parallel programming

GPU

deep learning

pony.ai

Develops autonomous driving technology solutions

About pony.ai

Pony.ai develops technology for autonomous driving, focusing on creating systems that can operate vehicles without human intervention. Their main product, the "virtual driver," is tested extensively in various driving conditions to ensure reliability. This technology is utilized in three main areas: Robotaxi services for passengers, Robotruck services for logistics and freight, and Personally Owned Vehicles for individual users. Unlike many competitors, Pony.ai tailors its solutions to meet the needs of different customer segments, including everyday travelers and commercial logistics companies. The company's goal is to advance autonomous mobility, making it accessible and efficient for everyone.

Fremont, CaliforniaHeadquarters

2016Year Founded

$718.8MTotal Funding

IPOCompany Stage

Robotics & Automation, Automotive & TransportationIndustries

501-1,000Employees

Benefits

Health Insurance

401(k) Retirement Plan

401(k) Company Match

Life Insurance

Paid Vacation

Parental Leave

Disability Insurance

Risks

Competition from Baidu in Hong Kong may impact Pony.ai's market share.

Regulatory changes in China could challenge Pony.ai's compliance and operations.

Tesla's expansion into China poses a competitive threat to Pony.ai.

Differentiation

Pony.ai's 'virtual driver' technology is a leader in autonomous driving innovation.

The company operates across Robotaxi, Robotruck, and Personally Owned Vehicles units.

Pony.ai collaborates with automotive manufacturers for seamless technology integration.

Upsides

Pony.ai plans to expand its robotaxi fleet to over 1,000 by 2025.

The company is introducing driverless services at Hong Kong International Airport.

Pony.ai received over $223 million in strategic investments in November 2024.

Land your dream remote job 3x faster with AI

Try Jobo Free