Job Description

Engineer, Data Science

KLA is seeking a motivated Engineer, Data Science to join our team. This role is perfect for a qualified 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.

📌 Engineer, Data Science (Chennai)
🏢 KLA
📍 Chennai