← Back to opportunities

Principal Machine Learning Engineer

📍 Location
Sunnyvale
⏰ Job Type
Full-time
📅 Posted
June 01, 2026

About the Role

**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...

Ready to Join Through a Referral?

Apply now and get connected directly with the hiring team

Apply for this Position