AI Frameworks Engineer

Intel

6 - 11 years

19 - 25 Lacs

bengaluru

Posted:1 day ago| Platform:

Apply

Skills Required

python advanced c hw tensorflow computer architecture jax image processing algorithms cuda c++ natural language processing gpu programming machine learning opengl artificial intelligence deep learning computer vision pytorch opencv flash ml

Work Mode

Work from Office

Job Type

Full Time

Job Description

We are looking for a dynamic and passionate contributors to work on Intels next generation GPUs.
Day-to-day work involves working on Open-source AI Frameworks such as PyTorch, Tensorflow & JAX and enabling cutting edge AI workloads (like LLMs) 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 software stack.

Role and responsibilities for this role includes:

Design and develop SW features for AI frameworks - both HW-agnostic and HW-aware, especially in PyTorch device plugin and 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
Working with Open-source community to support Intel GPUs and upstream code.

Qualifications:

BTech, MS/MTech & PhD 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 GPU programming.
Experience in developing machine learning kernels such as GEMM, Convolution, Flash attention, MoE etc.
Hands on experience in one of the frameworks such as PyTorch, Tensorflow or JAX.
Practical knowledge working on deep learning models such as NLPs & LLMs
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
Knowledge of compiler algorithms for heterogeneous system and Fuser optimizations.

More Jobs at Intel

Software QA Test Engineer

Bengaluru

3 - 6 yrs

INR 12 - 17 Lacs

Partner Sales Account Manager

Bengaluru

6 - 10 yrs

INR 18 - 22 Lacs

Senior SerDes PHY System Engineer

Bengaluru

5 - 10 yrs

INR 12 - 17 Lacs

Windows Wi-Fi Driver Engineer

Bengaluru

3 - 7 yrs

INR 10 - 14 Lacs

GPU Software Development Engineer

Bengaluru

7.0 - 11.0 yrs

INR 9 - 12 Lacs

Mock Interview

Practice Video Interview with JobPe AI

Start Python Interview

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.