Principal Machine Learning Engineer

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