Note: The job is a remote job and is open to candidates in USA. Motional is a driverless technology company developing autonomous vehicles and transportation solutions. The Staff Engineer Tech Lead Manager will build and lead a Machine Learning Acceleration team, defining strategy and developing solutions to accelerate model training and enable rapid hot-patching. The role combines hands-on technical leadership, people management, cross-functional collaboration, and performance measurement.
Responsibilities
- Build, lead, and manage a high-performing team of ML and infra engineers focused on acceleration
- Provide technical guidance, mentorship, and career development opportunities to team members
- Foster a collaborative and inclusive team environment
- Define team goals, priorities, and roadmap in alignment with company objectives
- Define the technical vision and strategy for ML acceleration across the organization
- Identify and evaluate cutting-edge technologies and methodologies for speeding up ML training, including but not limited to data pipeline optimization, large scale distributed training, data loader optimization, hardware acceleration, and model optimization techniques
- Design, develop, and implement scalable and efficient ML acceleration solutions
- Collaborate closely with ML research, ML Training platform, and product teams to understand their needs and integrate acceleration solutions seamlessly
- Communicate complex technical concepts and strategies to both technical and non-technical stakeholders
- Act as a technical expert and advocate for ML acceleration initiatives across the company
- Regularly measure and report on the impact of acceleration efforts
- Continuously seek opportunities for further optimization and innovation
Skills
- Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience
- 8+ years of experience in software engineering, with at least 5+ years focused on Machine Learning systems and model performance optimization
- 3+ years of experience in a technical lead or management role, with a proven track record of building and leading high-performing teams
- Strong understanding of distributed systems and cloud computing platforms (e.g., AWS, GCP, Azure)
- Deep expertise in ML frameworks such as PyTorch or JAX
- Proficiency in performance profiling and optimization techniques for Deep Neural Networks
- Knowledge of MLOps principles and practices
- Exceptional leadership, communication, and interpersonal skills
- Ability to attract, hire, and retain top engineering talent
- Proven ability to drive complex technical projects from conception to completion
- Strong problem-solving skills and a proactive, results-oriented mindset
- Ability to thrive in a fast-paced, dynamic environment
- Extensive experience with large-scale ML model training and deployment, ideally in a production environment
- Strong programming skills in Python; C++ experience is a plus
Benefits
- Medical, dental, and vision insurance
- 401k with a company match
- Health savings accounts
- Life insurance
- Pet insurance
- This role may include additional forms of compensation such as a bonus or company equity
Company Overview