Machine Learning Engineer Intern, ML Runtime & Optimization (Spring 2026, Master/PhD) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Pony.ai · 1 day ago

Machine Learning Engineer Intern, ML Runtime & Optimization (Spring 2026, Master/PhD)

positionFremont, CA
timeInternship
remoteOnsite
seniorityIntern
money$7,000/mo - $10,000/mo
workcardStart in 2026 Spring

Pony.ai is a global leader in autonomous mobility, recognized for its innovative technologies and services. They are seeking a Machine Learning Engineer Intern to develop technologies that enhance the training and inference of AI models used in autonomous driving systems.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesTransportation
check
H1B Sponsor Likelynote

Responsibilities

Performing in-depth analysis and optimization to model training and deployment to achieve the state of art in performance and efficiency in autonomous driving
Work across the entire AI framework/compiler stack (e.g. Torch, CUDA and TensorRT), support model development and prototype key deep learning algorithms
Analyze the tradeoffs between performance, cost and energy for autonomous driving
Collaborating closely with diverse groups in Pony.ai to influence the next-generation compute platform HW and SW design
Research the latest model architectures, programming models and hardware

Qualification

C/C++ programmingPython programmingGPU optimizationCUDAMachine learningDeep learningModel optimizationCommunicationCross-functional collaboration

Required

Currently pursuing a Masters or PhD program or a related discipline
Strong programming skills in C/C++ or Python
Solid understanding of CPU or GPU execution model, e.g. threads, registers, cache, memory, cost and performance trade-off, etc
Experience in benchmarking, profiling and validating performance
Strong communication skills and ability to work cross-functionally between software and hardware teams

Preferred

Experience with parallel programming: CUDA, ROCm, Triton, Cutlass, etc
Experience in computer vision, image processing, machine learning and deep learning
Experience in model optimization techniques such as quantization, pruning, etc
Experience in optimizing the utilization of compute resources, identifying and resolving compute and data flow bottlenecks
Strong knowledge of software design, programming techniques and algorithms
Strong knowledge of common deep learning frameworks and libraries
Strong knowledge on system performance, GPU optimization or ML compiler

Company

Pony.ai

twittertwittertwitter
Glassdoor3.4
company-logo
Pony.ai develops autonomous driving technology for vehicles that operates using artificial intelligence and machine learning.
dateFounded in 2016
location
Fremont, California, USA
people-size1001-5000 employees
web-link
https://www.pony.ai

H1B Sponsorship

Pony.ai has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (17)
2024 (13)
2023 (32)
2022 (44)
2021 (47)
2020 (24)

Funding

Current Stage
Public Company
Total Funding
$1.34B
Key Investors
NEOM Investment FundGAC Toyota MotorOntario Teachers' Pension Plan
2025-08-18Corporate Round· $12.9M
2024-11-27IPO
2023-10-24Series D· $100M

Leadership Team

leader-logo
James Peng
Founder and CEO
linkedin

Recent News

South China Morning Post
How Chinese robotaxi giants are steering the Middle East towards a driverless future
2026-01-03
36kr.com
Autonomous Driving Year-End Review 2025: Emerging from "Adolescence", Shifting from Technological Showmanship to Global Implementation
2025-12-31
Benzinga.com
'Great Potential,' Uber CEO Dara Khosrowshahi Says Of Japan, As He Targets '10-Plus Markets' Despite Robotaxis Costing About $150K Each
2025-12-28
Company data provided by crunchbase