We use cookies. Find out more about it here. By continuing to browse this site you are agreeing to our use of cookies.

Job posting has expired

Back to search results

Research Scientist Internship, ML Foundation Model

paid time off, tuition reimbursement, 401(k)
United States, California, Milpitas
December 08, 2023
Base Pay Range: $26.00 - $48.00 based on pursuit of a Masters and Ph.D. Primary Location: USA-CA-Milpitas-KLA KLA's total rewards package for employees may also include participation in performance incentive programs and eligibility for additional benefits identified below. Interns are eligible for some of the benefits identified below. Our pay ranges are determined by role, level, and location. The range displayed above reflects the minimum and maximum pay for this position in the primary location identified in this posting. Actual pay depends on several factors, including location, job-related skills, experience, and relevant education level or training. If applicable, your recruiter can share more about the specific pay range for your preferred location during the hiring process.

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.


KLA has always had a close relationship with physics and data. Our optical and electron beam inspection and measurement tools use cutting edge physics models, both for hardware design and as part of their algorithms. AI, including several traditional machine learning techniques and deep learning are routinely used to process this data to meet application requirements. The AI & Modeling Center of Excellence was setup with the mission of advancing KLA's traditional strengths in physics and data and providing implementation solutions for multiple KLA Inspection and Metrology products targeted at the semiconductor manufacturing industry. The AI & Modeling Center of Excellence is part of the company's Central Engineering organization providing product development expertise in a critical area for a wide variety KLA products. As a part of this group, you will be part of a world class team of physicists, HPC system designers, machine learning and application engineers who build cutting edge solutions for modeling complex imaging techniques and semiconductor processes. You will also work with a data scientists and AI infrastructure engineers whose mission is to build and scale machine learning based solutions for our semiconductor customers. We are looking for engineers in a few different fields. If you are passionate about Physics Modeling, High Performance Computing - HPC (including GPU), Machine Learning, Deep Learning, Data Sciences, or cutting-edge Cloud technologies - this is the place for you!

Job Description/Preferred Qualifications

We are looking for an internship candidate for summer 2023 (or summer 2024) OR part-time in the Fall, who is passionate on Machine Learning and Deep Learning research for image processing and computer vision applications. The ideal candidate is expected to have strong background on image processing algorithms, and also deep understanding on Machine Learning and Deep Learning theory.

The possible tasks that the candidate will be working on includes, but not limited to:

  • Understand and derive the algorithm from latest literatures and/or conferences.
  • Implement the state-of-art Machine Learning/Deep Learning models.
  • Design and/or Improve models for a specific application.
  • Design new Deep Learning modules.
  • Optimize and tune a Deep Learning Model.
  • Evaluate and validate the performance of deep learning models.
  • Perform A/B Test between models.
  • Tackle physical problems via machine learning or deep learning.
  • Communicate idea/finding to team members.

Qualifications/Education Desired

  • PhD (strongly preferred) OR Master's Level Degree candidate graduating in one year, in Electrical Engineering, Computer Science, Statistics or related quantitative fields.
  • Machine Learning/Deep Learning publication is required.
  • Experience on ML/DL Foundation modeling is required.
  • Research or industrial experience on applying Machine Learning/Deep Learning on real problems.
  • Experience in designing, validating and testing complex learning models.
  • Proficiency in Python; familiar with Deep Learning framework (e.g., TensorFlow, Caffe, PyTorch, etc).

Minimum Qualifications

  • Students working towards a PhD (preferred) or MS Level Degree.

The company offers a total rewards package that is competitive and comprehensive including but not limited to the following: 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, and family care and bonding leave.

KLA is proud to be an Equal Opportunity Employer. We do not discriminate on the basis of race, religion, color, national origin, sex, gender identity, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other status protected by applicable law. We will ensure that qualified individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us at to request accommodation.