KLA · 4 months ago
AI/ML Platform Engineering Lead
Ann Arbor, MI
Full-time
Onsite
Senior Level
$135K/yr - $229K/yr
5+ years expKLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. As an AI/ML Platform Engineering Lead, you’ll architect and scale MLOps infrastructure, enabling secure and reliable deployment of machine learning models across environments, while mentoring engineers and collaborating with cross-functional teams.
ElectronicsInformation TechnologyManufacturing
Responsibilities
Design, deploy and scale AI/ML platforms for model training and inference
Implement CI/CD pipelines for models and data workflows, ensuring reproducibility and compliance
Develop tools and frameworks that empower ML engineers to move from prototype to production seamlessly
Collaborate with cross-functional teams—including hardware and software—to integrate AI into instruments and factory workflows
Establish observability and governance for model performance, drift detection, and reliability
Mentor engineers and champion best practices in platform engineering and MLOps
Qualification
Required
Strong programming skills in Python; experience with ML frameworks (PyTorch, TensorFlow, Keras)
Expertise in MLOps tools (MLflow, Kubeflow), container orchestration (Kubernetes, Docker), and infrastructure automation (Terraform)
Experience deploying models in secure or disconnected environments (air gapped, on-prem)
Familiarity with cloud platforms (AWS, Azure, GCP) and hybrid architectures
Proven ability to lead technical design discussions and mentor engineering teams
Requires a minimum of 5 years of related experience with a Bachelors degree, or 3 years of experience and a Masters Degree
Benefits
Medical
Dental
Vision
Life
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
Paid company holidays
Family care and bonding leave
Company
KLA
Kla creates tools and services that promote innovation in the electronics industry.
10001+ employees
H1B Sponsorship
KLA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (343)
2024 (218)
2023 (191)
2022 (277)
2021 (200)
2020 (226)
Funding
Current Stage
Late StageLeadership Team
Lisa Grunwald
FISH Philosophy CEO
Rick Wallace
President & CEO
Recent News
news.com.au — Australia’s leading news site for latest headlines
2025-08-01
2025-08-01
Company data provided by crunchbase