AI Software Solutions Engineer (AI Frameworks, Workloads)

Job Description

We are looking for a dynamic software engineer to design, develop and optimize AI frameworks for training and inference on Intel Habana deep learning accelerators. In this role, you will work with a cross-geo team on enabling and optimizing state of the art deep learning models with a specific focus on the PyTorch framework. The roles and responsibilities that you would need to carry out may include the following:
Design and develop SW techniques for AI frameworks - both HW-agnostic and HW-aware Contribute to enhancing and extending the Training and Inference capabilities in the Software stack. Profile deep learning inference and training workloads and identify optimization opportunities in the software stack.

Qualifications

BTech, MS or PhD in CS or related fields with an overall experience of 10 to 15 years

Programming skills in Advanced C++, Python and parallel programming ski...