Staff Engineer Tech Lead Manager, ML Acceleration

Motional

$172K — $229K *
US-AnywhereRemote in United States
Information Technology
8 - 10 years of experience
Job Overview by Ladders

Qualifications

  • Bachelor's degree in Computer Science or related field; equivalent practical experience considered
  • 8+ years in software engineering, focusing on Machine Learning for at least 5 years
  • 3+ years in a technical lead or management role with a strong team-building record
  • Experience in large-scale ML training and deployment within production environments
  • Strong understanding of distributed systems and cloud platforms (AWS, GCP, Azure)
  • Deep expertise in ML frameworks like PyTorch or JAX
  • Proficiency in Python programming; familiarity with C++ is a plus
  • Knowledge of MLOps principles

Responsibilities

  • Build and manage a high-performing team of ML and infrastructure engineers
  • Provide mentorship and career development for team members
  • Set team goals and priorities aligned with company objectives
  • Define the technical vision for ML acceleration across the organization
  • Explore and implement cutting-edge technologies for ML training speed-up
  • Collaborate with product and research teams for seamless integration of solutions
  • Measure the impact of acceleration efforts and pursue ongoing optimization

Benefits

  • Medical, dental, and vision coverage
  • 401k plan with company match
  • Health savings accounts
  • Life and pet insurance
  • Opportunities for equity compensation
Full Job Description
Mission Summary:

We are seeking an experienced and visionary Principle Level Tech Lead Manager to build and lead our new Machine Learning (ML) Acceleration team. This pivotal role will drive the strategy, development, and execution of initiatives aimed at significantly accelerating ML model training. The ultimate goal is to drastically reduce the development cycle for new ML models and enable rapid hot-patching for issues within our deployed autonomous vehicle services.

You will be a hands-on leader, blending deep technical expertise in ML systems and performance optimization with strong leadership and people management skills. You will recruit, mentor, and grow a high-performing team of engineers, fostering a culture of innovation, collaboration, and continuous improvement.

What you'll be doing:
  • Team Leadership & Management:
    • 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.
  • Technical Strategy & Execution:
    • 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.
  • Cross-functional Collaboration:
    • 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.
  • Impact & Measurement:
    • Regularly measure and report on the impact of acceleration efforts.
    • Continuously seek opportunities for further optimization and innovation.

What we're looking for:
  • Experience:
    • 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.
    • Extensive experience with large-scale ML model training and deployment, ideally in a production environment.
    • Strong understanding of distributed systems and cloud computing platforms (e.g., AWS, GCP, Azure).
  • Technical Skills:
    • Deep expertise in ML frameworks such as PyTorch or JAX.
    • Proficiency in performance profiling and optimization techniques for Deep Neural Networks.
    • Strong programming skills in Python; C++ experience is a plus.
    • Knowledge of MLOps principles and practices.
  • Leadership Skills:
    • 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.


The salary range for this role is an estimate based on a wide range of compensation factors including but not limited to specific skills, experience and expertise, role location, certifications, licenses, and business needs. The estimated compensation range listed in this job posting reflects base salary only. This role may include additional forms of compensation such as a bonus or company equity. The recruiter assigned to this role can share more information about the specific compensation and benefit details associated with this role during the hiring process.

Candidates for certain positions are eligible to participate in Motional's benefits program. Motional's benefits include but are not limited to medical, dental, vision, 401k with a company match, health saving accounts, life insurance, pet insurance, and more.

Salary Range

$172,000-$229,000 USD

Similar Jobs

More Jobs at Motional

More Information Technology Jobs

Find similar Staff Engineer Tech Lead Manager, ML Acceleration jobs: