Principal Machine Learning Engineer

Job description

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 model training. Collaborate with data scientists, ML engineers, and software developers to enable seamless workflows from research to production. Drive efficiency in large-scale distributed training and data processing pipelines. Establish best practices for reli...