Lead AI Platform Engineer - Remote

Overview

As a Lead AI Platform Engineer at AssemblyAI, you will drive the technical direction of our data and AI training infrastructure team, shaping the vision and capabilities of our AI research platform.

In Short

• Lead the technical direction of AI infrastructure.
• Enhance and scale infrastructure for training AI models.
• Design scalable data platforms optimized for AI workloads.
• Build efficient data pipelines using GCP services.
• Implement cost-effective storage solutions for ML.
• Participate in on-call rotation for system reliability.
• Lead adoption of cutting-edge ML tools and frameworks.
• Streamline workflows and enhance tooling.
• Identify and eliminate bottlenecks in training pipelines.
• Collaborate closely with research teams and leadership.

Requirements

• 8+ years in AI/ML Infrastructure or related roles.
• 3+ years in a startup environment.
• Strong proficiency in Python and SQL.
• Deep expertise with GCP services.
• Experience with distributed processing frameworks.
• Familiarity with workflow orchestration tools.
• Understanding of distributed training systems.
• Experience with experiment tracking and training tooling.
• Ability to thrive in a fast-paced environment.

Benefits

• Competitive salary and equity options.
• Fully remote work environment.
• Opportunity to work with leading AI experts.
• Flexible working hours.
• Support for professional development.
• Inclusive company culture that values diversity.
• Access to cutting-edge technology and tools.
• Health and wellness benefits.
• Collaborative and innovative work environment.
• Contributions to meaningful AI projects.