Principal Machine Learning Engineer

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 reliability, scalability, and performance across the AI / ML platform.
• Provide technical leadership and mentorship, guiding teams on platform design, architecture decisions, and emerging technologies.
• Partner with cross-functional stakeholders to align platform capabilities with business needs and strategic AI initiatives.

Your Skills & Abilities (Required Qualifications) :

What Will Give You a Competitive Edge (Preferred Qualifications) :

Compensation : The compensation information is a good faith estimate only. It is based on what a successful applicant might be paid in accordance with applicable state laws. The compensation may not be representative for positions located outside of New York, Colorado, California, or Washington.

Work Location : This role is based remotely but if you live within a 50-mile radius of Atlanta, Austin, Detroit, Warren, Milford or Mountain View, you are expected to report to that location three times a week, at minimum.

#J-18808-Ljbffr