AI Engineer (E)

Job Description

KLA is seeking a skilled AI Engineer to join the Services AI Engineering team, focusing on creating robust learning systems, fine-tuning models, and driving advanced research and development.

Requirements

• Master’s degree and above in AI, Machine Learning, Computer Science, Engineering, or a related field.
• Proficiency in model fine-tuning techniques for large-scale models and tailoring them to specific tasks.
• Experience with inference optimization techniques (e.g. quantization, pruning, hardware-specific optimizations)
• Hands-on experience in machine learning, natural language processing, and AI technologies with contributions to open-source AI/ML projects.
• Strong programming skills in Python and familiarity with ML libraries (Ex. TensorFlow, Pytorch).
• Familiarity with MLOps practices and tools for deploying and managing models in production.
• Prior experience in a similar industry is a plus.
• Excellent problem-solving and analytical skills.
• Strong communication and teamwork abilities.

Responsibilities

• Research emerging AI technologies focused on contextual classification, multimodal models, and AI databases.
• Design, develop, and deploy proof of concepts from research.
• Collaborate with cross-functional teams to integrate models into existing systems and workflow.
• Participate in AI research, systems architecture discussions, and code reviews.
• Conduct data engineering and analysis to improve model accuracy and efficiency.
• Fine-tune models based on feedback and performance metrics.

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