AI Engineer

KLA • Ann Arbor, MI

2d

About The Position

We are seeking a skilled AI Engineer to join our Services AI Engineering team. You will focus on creating robust online and offline learning systems, fine-tuning models for optimal performance, developing evaluation strategies and ensuring efficient model inference. The ideal candidate will have a strong background in reinforcement learning theory and application, experience with model optimization techniques and a passion for driving advanced research and development for KLA use cases.

Requirements

• Master's degree without experience OR Bachelor's degree and a minimum of 2 years' of related experience; or equivalent work experience

Nice To Haves

• 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
• Prior experience in a similar industry is a plus

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 our 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

• medical
• dental
• vision
• life
• 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
• paid company holidays
• family care and bonding leave

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level