Job Description

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.
/Preferred Qualifications
Core Responsibilities

1. Implement and maintain PyTorch-based training and inference pipelines.

2. Develop modular, reusable code for data handling, model serving, and evaluation.

3. Optimize code to run faster and keep accuracy

4. Optimize data transfer between systems (e.g., Windows Linux), including gRPC or other protocols.

5. Ensure code quality through testing, documentation, and CI/CD practices.

6. Support deployment of models into production environments or internal tools.

7. Collaborate with the algorithm expert to translate research into scalable solutions.
Minimum Qualifications
Qualifications/Requirements:

Required: MS/MTech degree with 5 or more years of Algorithm development experience.

Required: Strong Python and PyTorch engineering skills.

Required: Knowledge of software testing, version control, and deployment best practices.

Preferred: Experience with data serialization and transfer protocols (e.g., gRPC, protobuf, shared memory).

Preferred: Familiarity with cross-platform development and system integration.

Preferred: Math backgroundWe 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 talent.acquisition@kla.com to confirm the person you are communicating with is an employee. We take your privacy very seriously and confidentially handle your information.