AI/ML Platform Engineering Lead

Join KLA's AI/ML engineering team to build the backbone that powers production AI. As an AI/ML Platform Engineering Lead, you'll architect and scale our MLOps infrastructure, enabling secure, reliable, and fast deployment of machine learning models across cloud and air gapped environments.

Requirements

• 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

Benefits

• medical
• dental
• vision
• life
• 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