Job Details:

Job Description: 

We are looking for a dynamic software engineer to design, develop and optimize AI frameworks for training and inference on Intel Gaudi (https://habana.ai/) 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 deep-learning frameworks. The roles and responsibilities that you would need to carry out may include the following: � Profile and enable deep learning inference and training workloads on Intel Gaudi DL accelerators. Identify and fix functional and convergence issues, identify performance optimization opportunities and drive implementation of these optimizations. � Design and develop SW techniques for AI frameworks - both HW-agnostic and HW-aware. � Contribute to enhancing and extending the Training and Inference capabilities of DL accelerator Software stack.

Qualifications:

� BTech, MS or PhD in CS or related fields with an overall experience of 3-10 years. � Programming skills in Python, C++, and parallel programming skills. � Previous exposure to Deep Learning (DL) frameworks such as PyTorch, Tensorflow, JAX. � Familiarity with Generative AI models and LLMs (such as Llama, GPT3, Mixtral, Falcon etc.). � Exposure to working on open-source DL projects (such as HuggingFace transformers, accelerate, peft ; Facebook research Fairseq, Fairscale, Detectron etc.). � Working knowledge of operators in Pytorch/Tensorflow and Understanding of low-level kernels. � Understanding of computer architecture and HW-SW optimization techniques. � Experience working on frameworks/platforms that have gone to production. � Effective communication skills and experience with working in a cross-geo setup. Preferred � Experience of contributing and maintaining large open-source projects. � Familiarity with DL models in computer-vision, NLP, Recommendation systems, object detection and segmentation. � Knowledge of compiler algorithms for heterogeneous systems. � Knowledge of open-source compiler infrastructure like LLVM, MLIR, XLA HLO or gcc.

          

Job Type:

Experienced Hire

Shift:

Shift 1 (India)

Primary Location: 

India, Bangalore

Additional Locations:

Business group:

The Data Center & Artificial Intelligence Group (DCAI) is at the heart of Intel’s transformation from a PC company to a company that runs the cloud and billions of smart, connected computing devices. The data center is the underpinning for every data-driven service, from artificial intelligence to 5G to high-performance computing, and DCG delivers the products and technologies—spanning software, processors, storage, I/O, and networking solutions—that fuel cloud, communications, enterprise, and government data centers around the world.

Posting Statement:

All qualified applicants will receive consideration for employment without regard to race, color, religion, religious creed, sex, national origin, ancestry, age, physical or mental disability, medical condition, genetic information, military and veteran status, marital status, pregnancy, gender, gender expression, gender identity, sexual orientation, or any other characteristic protected by local law, regulation, or ordinance.

Position of Trust

N/A

Work Model for this Role

This role will be eligible for our hybrid work model which allows employees to split their time between working on-site at their assigned Intel site and off-site. In certain circumstances the work model may change to accommodate business needs.