Engineer, Data Science

Engineer, Data Science

KLA is seeking a motivated Engineer, Data Science to join our team. This role is perfect for a professional passionate about analyzing, designing, developing, testing, and deploying software solutions that enable the deployment of AI frameworks in heterogeneous clusters, overcoming existing limitations.

Responsibilities

• Analyze existing software solutions for training and deployment and identify limitations.
• Design, develop, test, and deploy software solutions that address identified limitations and enable deployment in a heterogeneous cluster.
• Collaborate with a global team in developing such solutions.

Skills

• Active work in AI frameworks like TensorFlow, PyTorch, ONNX (preferred).
• Hands-on experience working with container infrastructure like Docker or Kubernetes (preferred).
• Background in modern programming languages with Rust (preferred).
• Experience with building open-source operating systems and software stacks (preferred).
• Experience with writing performance-sensitive Python programs using frameworks like Triton, Polars, etc. (preferred).

Qualifications

• Bachelor's or Master's degree in Engineering.