Deep Learning Engineer — Vision & GenAI

KLA-Belgium; Milpitas, Northern (CA, KY); Hybrid; USD 136,000 - 232,000; Full time

Benefits offered by this job

Medical benefits

401K

ESPP

Tuition reimbursement

Career growth

Financial planning

Wellness program

Paid time off

Job summary

KLA in Milpitas, CA is seeking a full-time Deep Learning Algorithm Engineer passionate about pioneering DL, foundation models, and GenAI for image processing in the semiconductor process control business.

The role requires hands-on experience with DL model development, training, optimization, and deployment, with emphasis on performance and scalability.

PhD in EE/CS or MS with relevant experience is preferred; strong Python and C++ skills, and experience with PyTorch/TensorFlow/JAX are expected.

Qualifications

• Ph.D. or equivalent in EE/CS or related field with strong DL focus.
• Experience applying DL/GenAI to real-world problems with impactful results.
• Expertise in Vision Foundation Models, VFM, or Vision-Language Models.
• Proficient in Python and C++, Java, Rust, or Go.
• Experience with DL model optimization for mixed or reduced precision.
• Publications or patents in DL or GenAI preferred.

Responsibilities

• Understand and apply state-of-the-art DL and GenAI models.
• Design and implement DL/GenAI models for image processing.
• Train and tune models on domain datasets.
• Optimize models for throughput and deployment.
• Collaborate with peers and present results.

Skills

Python

C++

Java

Rust

Go

Education

Ph.D. in Electrical Engineering, Computer Science, or related quantitative fields

Master’s degree with 3 years of related work experience

Tools

PyTorch

TensorFlow

JAX

Job description

KLA in Milpitas, CA is seeking a full-time Deep Learning Algorithm Engineer passionate about pioneering DL, foundation models, and GenAI for image processing in the semiconductor process control business.

The role requires hands-on experience with DL model development, training, optimization, and deployment, with emphasis on performance and scalability.

PhD in EE/CS or MS with relevant experience is preferred; strong Python and C++ skills, and experience with PyTorch/TensorFlow/JAX are expected.