Senior Software Engineer - AI/ML - KLA

Key Responsibilities

• Design, develop, and optimize AI/ML algorithms and models for semiconductor manufacturing analytics.
• Implement high‑throughput data pipelines and inference services using Python and C++.
• Leverage GPU acceleration with CUDA to achieve real‑time performance on large datasets.
• Collaborate with cross‑functional teams to integrate ML solutions into existing wafer‑inspection and metrology platforms.
• Conduct model validation, performance tuning, and continuous improvement of deployed solutions.

Requirements

• 5+ years of professional software development experience, with strong proficiency in Python and C++.
• Hands‑on experience building and deploying deep‑learning models using TensorFlow or PyTorch.
• Expertise in GPU programming (CUDA) and performance optimization for large‑scale data processing.
• Solid understanding of machine‑learning concepts, data preprocessing, and model evaluation.
• Experience working in a Linux development environment and using version‑control systems (e.g., Git).