AI Infrastructure Engineer

Job Description

Join KLA, a global leader in electronics for semiconductor manufacturing, and work on cutting-edge AI products for semiconductor manufacturing challenges.

Requirements

• Educational background in Computer Science, Mathematics, or a related field.
• Proficiency in Python and Machine Learning (ML) frameworks such as PyTorch, TensorFlow, and Keras.
• Hands-on experience with Large Language Models and related tools.
• Strong problem-solving abilities and experience with statistical analysis.
• Ability to work collaboratively with fellow team members and lead multiple projects simultaneously.

Responsibilities

• Lead the development and implementation of AI products, ensuring they meet project requirements and business goals.
• Help the team build infrastructure and deploy modern MLOps techniques to our customer sites.
• Rapidly prototype and iterate on machine learning applications, demonstrating large language models (LLMs) for various use cases.
• Stay updated with the latest ML/AI research and integrate new techniques into our products.
• Select appropriate datasets, perform data preprocessing, and ensure data quality for training models.
• Conduct machine learning tests, perform statistical analysis, and fine-tune models for efficient performance.
• Work closely with multi-disciplinary teams, including data scientists, software engineers, and product managers, to deliver high-quality AI solutions.
• Guide and mentor junior team members, encouraging a culture of continuous learning and improvement.

Benefits

• Performance incentive programs
• Medical, dental, vision, and life insurance (plus other voluntary benefits)
• 401(k) with company matching
• Employee Stock Purchase Program (ESPP)
• Student debt assistance
• Tuition reimbursement program
• Development and career growth opportunities
• Financial planning benefits
• Wellness benefits (including EAP)
• Paid time off and paid company holidays
• Family care and bonding leave