AI Frameworks Engineer

About the role

Job Details:

Job Description:

The Role and Impact

Intel is building the open software stack that makes large language models run fast on Intel GPUs. You will join the engineering team that develops and optimizes LLM inference frameworks - working directly in vLLM and in the SW stack beneath it and in the upstream. You will write and tune compute kernels, reshape execution paths, hunt performance bottlenecks and submit your changes into widely used open-source projects. If you want to optimize how fast AI workloads run, this is that role.

Business group

The Data Center Group (DCG) is at the forefront of delivering cutting-edge technologies to power the world's data centers, enabling advanced computing solutions that drive innovation across industries. Within this group, our team is dedicated to developing AI frameworks and workloads that empower researchers and developers to create high-performance, scalable solutions for complex AI challenges. We aim to build tools and technologies that accelerate progress in AI, aligning with Intel's mission to advance global technology.

Key Responsibilities

- Design, develop, and optimize AI framework software for LLM inference on Intel GPUs, contributing upstream to external open-source frameworks such as vLLM and PyTorch.

- Develop and optimize compute kernels for transformer inference.

- Implement and tune distributed execution paths - tensor, pipeline, and data parallelism, collective communication, and asynchronous data movement within deep learning frameworks.

- Transform and optimize the computational-graph representation of neural network models, and develop machine learning primitives in mathematical and kernel libraries.

- Profile inference workloads to identify performance bottlenecks across the stack, quantify them, and propose and drive solutions together with the individual component teams that own each layer.

- Optimize code for a range of computing hardware backends, adapting implementations to the memory hierarchy, instruction set, and execution model of each target.

- Build and extend unit tests, correctness checks, and performance regression coverage so that framework and kernel work stays reliable as upstream moves.

- Collaborate with framework, compiler, and machine learning engineers across global teams, and engage with the open-source community through code review and technical discussion.

Qualifications:

Minimum Qualifications

- Bachelor's degree in AI, Computer Science, Computer Engineering, Mathematics, Physics, or a related field.

- 4+ years of relevant experience with a Bachelor's degree, or 3+ years with a Master's degree.

- Advanced expertise in Python programming.

- Hands-on experience with AI inference and large language models (LLMs).

- Demonstrated proficiency in PyTorch.

- Solid foundation in AI frameworks, machine learning algorithms, and software engineering practices.