Software Engineer (AI/ML) Intern

KLA•Milpitas, CA
•$22 - $43•Onsite

About The Position

KLA’s Measurement, Analytics and Control (MACH) division develops high-performance software platforms that support semiconductor manufacturing. The Software Engineer (AI/ML) Intern may work on platforms for large-scale software systems, data management, and machine-learning model training and inference solutions for KLA products. The role calls for strong analytical and problem-solving capabilities, a passion for software, and hands-on development experience. Familiarity with machine learning and deep learning is beneficial. Candidates should also demonstrate creative approaches to complex data, effective communication, teamwork, and motivation by team success.

Requirements

  • Be a master's-level degree student or a PhD candidate.
  • Have exceptional software development skills and hands-on experience in Python, C++, C#, or Java.
  • Have a deep conceptual understanding of multithreaded and multiprocess software systems.
  • Demonstrate strong analytical and problem-solving skills.
  • Demonstrate the ability to formulate creative solutions by analyzing complex data.
  • Have good communication skills.
  • Be a strong team player motivated by team success.

Nice To Haves

  • Familiarity with machine-learning and deep-learning solutions.
  • Hands-on experience with data structures and algorithms.
  • Experience with traditional machine-learning methods such as Random Forest, XGBoost, and Logistic Regression.
  • Experience with deep learning for regression, classification, and generative models.
  • Experience with TensorFlow or PyTorch, NumPy, scikit-learn, or other ML and DL frameworks.
  • Experience with distributed systems for data management.
  • Experience with distributed computing infrastructure.
  • Experience with cloud technologies for storage, containerization, and compute clusters.
  • Experience with GPU architectures and data management.

Responsibilities

  • Contribute to platforms for large-scale software systems.
  • Work on data management.
  • Develop or support machine-learning model training and inference solutions for KLA products.

Benefits

  • Medical, dental, vision, life, and other voluntary benefits (intern eligibility may vary).
  • 401(k) with company matching (intern eligibility may vary).
  • Employee stock purchase program (ESPP) (intern eligibility may vary).
  • Student debt assistance (intern eligibility may vary).
  • Tuition reimbursement program (intern eligibility may vary).
  • Development and career growth opportunities and programs (intern eligibility may vary).
  • Financial planning benefits (intern eligibility may vary).
  • Wellness benefits, including an employee assistance program (EAP) (intern eligibility may vary).
  • Paid time off and paid company holidays (intern eligibility may vary).
  • Family care and bonding leave (intern eligibility may vary).
  • Potential participation in performance incentive programs.
© 2026 Teal Labs, Inc
Privacy PolicyTerms of Service