Applied AI (Frameworks) Engineer

Opens intel.wd1.myworkdayjobs.com in a new tab

About This Role

  • We are looking for a dynamic and passionate senior contributor to work in Intel's Data Center and AI group (DCAI).
  • Day-to-day work involves working on Open source AI Frameworks such as PyTorch, Tensorflow, JAX etc.
  • The job role involves design and developing features for Intel' AI frameworks software stack.
  • You will be participating in enabling and optimizing state of the art Software stack for Intel's AI accelerators and next generation GPUs.
  • The roles and responsibilities that you would need to performance may include the following: • Design and develop SW features for AI frameworks - both HW-agnostic and HW-aware, especially in ML kernel development. • Enhance and extend the Deep learning training, and Inference capabilities in the Software stack. • Identifying optimization opportunities in the software stack to enhance performance of Deep learning workloads • Participate in discussions with Open-source community, involve in development, adopting upstream and Upstream software.

Qualifications

  • BTech or MS/MTech in CS, ECE or related fields with an overall experience of 6 to 12 years.
  • Proficient in Advanced C++ (C++ 14/17) and Intermediate skills of Python and parallel programming.
  • Experience in developing machine learning kernels such as GEMM, Convolution, Flash attention etc In depth and hands on experience in one of the frameworks such as PyTorch, Tensorflow or JAX Practical knowledge of Deep Learning models/LLMs for Vision / NLP Ability to debug complex issues in multi layered SW systems.
  • Understanding of SW integration in large open-source frameworks.
  • Strong understanding of computer architecture and HW-SW optimization techniques.
  • Experience in working on frameworks/platforms that have gone to production.
  • Effective communication skills and experience with working in a cross-geo teams.
  • Preferable Experience in developing and integrating CUTLASS or Triton based kernels in Large language models (LLMs).
  • Knowledge of compiler algorithms for heterogeneous system and Fuser optimizations.BTech or MS/MTech in CS, ECE or related fields with an overall experience of 6 to 12 years.
  • Proficient in Advanced C++ (C++ 14/17) and Intermediate skills of Python and parallel programming.
  • Experience in developing machine learning kernels such as GEMM, Convolution, Flash attention etc In depth and hands on experience in one of the frameworks such as PyTorch, Tensorflow or JAX Practical knowledge of Deep Learning models/LLMs for Vision / NLP Ability to debug complex issues in multi layered SW systems.
  • Understanding of SW integration in large open-source frameworks.
  • Strong understanding of computer architecture and HW-SW optimization techniques.
  • Experience in working on frameworks/platforms that have gone to production.
  • Effective communication skills and experience with working in a cross-geo teams.
  • Preferable Experience in developing and integrating CUTLASS or Triton based kernels in Large language models (LLMs).
  • Knowledge of compiler algorithms for heterogeneous system and Fuser optimizations.

Sourced directly from Intel’s career page

Your application goes straight to Intel.

Intel logo

Intel

India, Bangalore

Specialisation
Open roles at Intel
765 positions
Job ID
/job/India-Bangalore/Applied-AI--Frameworks--Engineer_JR0283585-1

Get matched to roles like this

Upload your resume once. We’ll notify you when matching roles open up.

Join talent pool — free

Similar Other roles