AI/ML Engineer - MLOps/SRE

Reposted 14 Days Ago
Be an Early Applicant
Chennai, Tamil Nadu, IND
In-Office
Mid level
Hardware
The Role
Build and operate production ML/LLM systems: design MLOps platform capabilities, standardized pipelines, model registry and artifact lineage; deploy using containers/Kubernetes and CI/CD; implement observability, SLOs, incident response, model/data monitoring, security controls, and enable teams with documentation and best practices.
Summary Generated by Built In

Company Overview

KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. Virtually every electronic device in the world is produced using our technologies. No laptop, smartphone, wearable device, voice-controlled gadget, flexible screen, VR device or smart car would have made it into your hands without us. KLA invents systems and solutions for the manufacturing of wafers and reticles, integrated circuits, packaging, printed circuit boards and flat panel displays. The innovative ideas and devices that are advancing humanity all begin with inspiration, research and development. KLA focuses more than average on innovation and we invest 15% of sales back into R&D. Our expert teams of physicists, engineers, data scientists and problem-solvers work together with the world’s leading technology providers to accelerate the delivery of tomorrow’s electronic devices. Life here is exciting and our teams thrive on tackling really hard problems. There is never a dull moment with us.

Group/Division

The Information Technology (IT) group at KLA is involved in every aspect of the global business. IT’s mission is to enable business growth and productivity by connecting people, process, and technology. It focuses not only on enhancing the technology that enables our business to thrive but also on how employees use and are empowered by technology. This integrated approach to customer service, creativity and technological excellence enables employee productivity, business analytics, and process excellence.

Job Description/Preferred Qualifications

We are seeking a hands-on AI/ML Engineer specializing in MLOps and Site Reliability Engineering (SRE) to build, operate, and continuously improve production-grade machine learning systems. In this role, you will partner with data scientists, data engineers, and software teams to standardize the ML lifecycle, improve reliability and performance, and enable rapid, safe delivery of models and AI services at scale. 

Key Responsibilities 

  • Production ML Platform & Tooling 

  • Design and implement reusable MLOps platform capabilities for training, deployment, and monitoring of ML/LLM systems. 

  • Build standardized pipelines for data validation, feature generation, training, evaluation, model packaging, and release. 

  • Own model registry, artifact storage, and metadata lineage to ensure reproducibility and auditability. 

  • Deployment Engineering & Release Safety 

  • Deploy models and AI services using containers and orchestration (e.g., Kubernetes) with robust rollout strategies (blue/green, canary, A/B). 

  • Create CI/CD workflows for ML code and pipelines, including automated tests, quality gates, and approval controls. 

  • Harden inference services for low latency and high throughput using caching, batching, autoscaling, and efficient model serving patterns. 

  • Reliability, Observability & Incident Response (SRE) 

  • Define and track service-level indicators (SLIs) and service-level objectives (SLOs) for ML services, pipelines, and data dependencies. 

  • Implement end-to-end observability: structured logging, metrics, tracing, dashboards, and alerting for both infrastructure and model behavior. 

  • Lead incident response and post-incident reviews; drive systemic fixes through runbooks, automation, and reliability engineering practices. 

  • Model & Data Monitoring 

  • Implement monitoring for model quality and data health: drift, bias, performance degradation, and data pipeline anomalies. 

  • Build automated feedback loops to trigger investigations, retraining workflows, and safe rollback when quality thresholds are breached. 

  • Security, Compliance & Governance 

  • Integrate security best practices: secrets management, least-privilege access (RBAC), network controls, and vulnerability scanning. 

  • Support compliance and governance requirements for model usage, data access, retention, and responsible AI practices. 

  • Collaboration & Enablement 

  • Partner with data science and engineering teams to translate business requirements into reliable, scalable ML solutions. 

  • Create developer-friendly documentation, templates, and internal best practices; mentor teams on MLOps and reliability standards. 

Required Qualifications 

  • Bachelor's degree in Computer Science, Engineering, Data Science, or a related field with 5+ years of relevant experience; OR a Master's/PhD with 3+ years of relevant experience. 

  • Proven experience deploying and operating ML models or AI services in production environments. 

  • Strong programming skills in Python and experience with common ML libraries and frameworks (e.g., PyTorch, TensorFlow, scikit-learn). 

  • Hands-on DevOps/SRE experience: CI/CD, infrastructure as code, containerization, and operational excellence. 

  • Experience with cloud platforms and managed services (Azure, AWS, or GCP) and building scalable, secure systems. 

  • Experience with Kubernetes and modern model serving patterns (REST/gRPC, async workers, batch/stream inference). 

  • Strong understanding of monitoring and observability (metrics, logs, traces) and incident management processes. 

  • Ability to communicate clearly with both technical and non-technical stakeholders, and to operate effectively in cross-functional teams. 

Preferred Qualifications 

  • Experience with ML platform tools such as MLflow, Kubeflow, Airflow, SageMaker, Vertex AI, or Azure Machine Learning. 

  • Experience with feature stores, data quality frameworks, and dataset/versioning tools (e.g., Feast, Great Expectations, DVC). 

  • Experience with distributed systems performance tuning (autoscaling, queueing, caching, load shedding). 

  • Experience implementing model monitoring for drift, bias, and quality (e.g., Evidently, whylogs, custom statistical checks). 

  • Knowledge of security and compliance patterns for enterprise AI (data classification, encryption, audit logging). 

  • Contributions to open-source projects, publications, or demonstrated technical leadership through talks/blogs. 

What Success Looks Like (First 6-12 Months) 

  • Standardized CI/CD and deployment patterns for ML services that reduce time-to-production while improving safety and reliability. 

  • Clear SLOs, dashboards, and alerts for critical AI services with measurable improvements in uptime, latency, and incident response. 

  • Automated monitoring and quality checks that detect drift and data issues early, with repeatable remediation workflows. 

  • Improved reproducibility and governance through consistent artifact tracking, lineage, and documentation. 

 

Note: Technology choices may vary by team needs; candidates should be comfortable learning and adapting to new tools. 

Minimum Qualifications

Doctorate (Academic) Degree and 0 years related work experience; Master's Level Degree and related work experience of 3 years; Bachelor's Level Degree and related work experience of 5 years

We offer a competitive, family friendly total rewards package. We design our programs to reflect our commitment to an inclusive environment, while ensuring we provide benefits that meet the diverse needs of our employees.

KLA is proud to be an equal opportunity employer

Be aware of potentially fraudulent job postings or suspicious recruiting activity by persons that are currently posing as KLA employees.  KLA never asks for any financial compensation to be considered for an interview, to become an employee, or for equipment. Further, KLA does not work with any recruiters or third parties who charge such fees either directly or on behalf of KLA. Please ensure that you have searched KLA’s Careers website for legitimate job postings.  KLA follows a recruiting process that involves multiple interviews in person or on video conferencing with our hiring managers.  If you are concerned that a communication, an interview, an offer of employment, or that an employee is not legitimate, please send an email to [email protected] to confirm the person you are communicating with is an employee. We take your privacy very seriously and confidentially handle your information.

Skills Required

  • Bachelor's degree in Computer Science, Engineering, Data Science, or related field with 5+ years experience OR Master's/PhD with 3+ years experience
  • Proven experience deploying and operating ML models or AI services in production environments
  • Strong programming skills in Python
  • Experience with common ML libraries and frameworks (PyTorch, TensorFlow, scikit-learn)
  • Hands-on DevOps/SRE experience: CI/CD, infrastructure as code, containerization, operational excellence
  • Experience with cloud platforms and managed services (Azure, AWS, or GCP)
  • Experience with Kubernetes and modern model serving patterns (REST/gRPC, async workers, batch/stream inference)
  • Strong understanding of monitoring and observability (metrics, logs, traces) and incident management processes
  • Ability to communicate clearly with technical and non-technical stakeholders and work in cross-functional teams
  • Ownership of model registry, artifact storage, and metadata lineage to ensure reproducibility and auditability

KLA Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about KLA and has not been reviewed or approved by KLA.

  • Retirement Support Retirement offerings include a 401(k) plan with company matching and financial planning support. Student debt assistance and related financial benefits reinforce long-term savings and security.
  • Equity Value & Accessibility Ownership programs include an Employee Stock Purchase Plan and broad-based RSU participation that extend equity beyond a narrow group. These elements complement competitive pay and bonuses to strengthen total rewards.
  • Leave & Time Off Breadth Time-off programs span paid time off, paid company holidays, and paid volunteer time. Family care and bonding leave and back-up care services add flexibility during life events.

KLA Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Milpitas, CA
10,001 Employees

What We Do

KLA develops industry-leading equipment and services that enable innovation throughout the electronics industry. We provide advanced process control and process-enabling solutions for manufacturing wafers and reticles. In close collaboration with leading customers across the globe, our expert teams of physicists, engineers, data scientists and problem-solvers design solutions that move the world forward.

Similar Jobs

Capco Logo Capco

Data Quality

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
India
6000 Employees

Capco Logo Capco

Data Engineer

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
2 Locations
6000 Employees

Capco Logo Capco

Business Analyst

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
India
6000 Employees

Capco Logo Capco

IRR - Regulatory Reporting

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
India
6000 Employees

Similar Companies Hiring

Red 6 Thumbnail
Aerospace • Hardware • Software • Virtual Reality • Defense
Orlando, Florida
186 Employees
Blissway Thumbnail
Computer Vision • Fintech • Hardware • Internet of Things • Machine Learning • Software • Transportation
Denver, CO
24 Employees
Fairly Even Thumbnail
Hardware • Robotics • Sales • Software • Hospitality
New York, NY
30 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account