remote
AI/ML Engineer - Systems - KLA
ML Engineer
Lead the design and deployment of AI/ML solutions for semiconductor manufacturing systems, leveraging Python, deep learning frameworks, and AWS cloud services to enhance process efficiency and product quality.
About the role
Key Responsibilities
- Develop and production‑grade machine learning models to predict and optimize semiconductor fabrication processes.
- Collaborate with cross‑functional teams to integrate AI/ML insights into real‑time control systems.
- Design scalable data pipelines and feature stores using AWS services (S3, Glue, SageMaker).
- Implement model monitoring, retraining workflows, and performance dashboards.
- Contribute to research and prototyping of novel deep learning architectures for defect detection and yield improvement.
Requirements
- BS/MS in Computer Science, Electrical Engineering, or related field.
- 3+ years of experience building production ML systems in Python.
- Strong knowledge of deep learning frameworks (PyTorch, TensorFlow) and AWS ML services.
- Experience with data engineering, ETL pipelines, and cloud infrastructure.
- Excellent problem‑solving skills and ability to communicate complex ideas to technical and non‑technical stakeholders.
Skills
pythonmachine learningdeep learningaws