Sr. AI Software Engineer

Job Description

Join KLA, a global leader in diversified electronics for the semiconductor manufacturing ecosystem, to develop AI products and solutions 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 build infrastructure and deploy modern MLOps techniques to customer sites
• Rapidly prototype and iterate on machine learning applications for various use cases
• Stay updated with the latest ML/AI research and integrate new techniques into products
• Conduct machine learning tests, perform statistical analysis, and fine-tune models for efficient performance
• Work closely with multi-disciplinary teams 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