Principal Machine Learning Engineer
Location
Sunnyvale, CA
Job Type
Full-time
Category
other-general
Posted
June 01, 2026
**Job Description**
_This role is based remotely but if you live within a 50-mile radius of [Austin, Detroit, Warren, Milford or Mountain View], you are expected to report to that location three times a week, at minimum._
**The Role:**
We are seeking a Principal AI Engineer to lead the design and advancement of our AI platform. You will play a key role in shaping the infrastructure that powers large-scale training and cloud inference. This includes accelerating training throughput, scaling multi-modal models, and enabling the next generation of AI-driven driving systems. We're tackling challenges across distributed training, training efficiency, DDP/FSDP, data processing pipelines, and Pytorch model optimization. This is a highly impactful position where your technical leadership will define how we scale AI to achieve autonomy.
**What You’ll Do:**
+ Architect, build, and optimize core AI/ML platform infrastructure to support massive-scale mo...
_This role is based remotely but if you live within a 50-mile radius of [Austin, Detroit, Warren, Milford or Mountain View], you are expected to report to that location three times a week, at minimum._
**The Role:**
We are seeking a Principal AI Engineer to lead the design and advancement of our AI platform. You will play a key role in shaping the infrastructure that powers large-scale training and cloud inference. This includes accelerating training throughput, scaling multi-modal models, and enabling the next generation of AI-driven driving systems. We're tackling challenges across distributed training, training efficiency, DDP/FSDP, data processing pipelines, and Pytorch model optimization. This is a highly impactful position where your technical leadership will define how we scale AI to achieve autonomy.
**What You’ll Do:**
+ Architect, build, and optimize core AI/ML platform infrastructure to support massive-scale mo...