Lead Machine Learning Engineer

KLA • Ann Arbor, MI

106d • $102,200 - $173,700

About The Position

Whether you've led teams or projects, passionate about solving complex problems and want to stay close to the tech, this is your opportunity to lead through innovation. Join us in solving one of the most sophisticated challenges in semiconductor manufacturing: detecting nanoscale defects with limited data and system access! You will lead from the front—designing, developing, and deploying ground-breaking AI models that push the boundaries of what's possible. We’re looking for someone who thrives with solving hard problems, guiding teams through influence, and staying hands-on with the latest in machine learning and MLOps. If you're ready to build the future of AI in manufacturing, we want to hear from you!

Requirements

• A solid base in Computer Science, Mathematics, or a related field.
• Deep expertise in Python and ML frameworks like PyTorch, TensorFlow, or Keras.
• Hands-on experience with LLMs and their practical applications.
• Sharp analytical skills and a curiosity about solving complex problems.
• The ability to communicate technical ideas clearly to diverse audiences.
• Shown success leading projects or teams—formally or informally.
• A collaborative attitude and the ability to juggle multiple priorities.

Nice To Haves

• Experience deploying ML models on cloud platforms (Azure, AWS, GCP).
• A background in AI/ML product development or similar roles.

Responsibilities

• Drive AI Product Development: Lead the design and implementation of AI solutions that meet real-world business needs.
• Build and Deploy Infrastructure: Help architect scalable systems and bring modern MLOps practices to life at customer sites.
• Prototype Rapidly: Experiment with LLMs and ML applications to uncover new use cases and deliver quick wins.
• Stay Ahead of the Curve: Integrate ground breaking ML/AI research into production-ready tools.
• Lead the Data Pipeline: Select, clean, and prepare datasets to ensure high-quality model training.
• Optimize Performance: Run experiments, analyze results, and fine-tune models for speed and accuracy.
• Collaborate multi-functionally: Work closely with data scientists, engineers, and product managers to deliver impactful solutions.
• Mentor Through Influence: Guide junior team members and develop a culture of learning—without formal management responsibilities.

Benefits

• Medical, dental, vision, life, and other voluntary benefits.
• 401(K) including company matching.
• Employee stock purchase program (ESPP).
• Student debt assistance.
• Tuition reimbursement program.
• Development and career growth opportunities and programs.
• Financial planning benefits.
• Wellness benefits including an employee assistance program (EAP).
• Paid time off and paid company holidays.
• Family care and bonding leave.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: Bachelor's degree
• Number of Employees: 5,001-10,000 employees