AI Frameworks Architect Full-time

Job Details:

Job Description:
Are you passionate about exploring computer architecture for AI? Are you excited about squeezing every cycle of performance through software optimizations? We are seeking world-class software developers and performance architects to join us and optimize open source AI frameworks, libraries and models on Intel accelerators. In this role, you will have opportunity to democratize AI by working with hundreds of thousands smart developers in open ecosystem and make a big technical impact in open source AI frameworks and libraries such as PyTorch, HuggingFace, JAX, vLLM etc.

What You’ll Be Doing
– Craft the design to get Intel accelerators/GPUs/CPUs optimized in open source AI frameworks.
– Performance modelling, profiling, debug, and code optimization with architectural knowledge of hardware
– Develop highly optimized deep learning kernels for Intel accelerators/GPUs/CPUs.
– Interact with multiple technologists in the company to influence architectures and optimize/customize software offerings.
– Explore and expand the boundary of innovative technologies for peak performance.

Qualifications:

You are good at:
– Master, PhD or equivalent in relevant fields (Computer Science, EE, Math)
– 12+ years of relevant working experience
– Strong programming skills in C, C++, and Python
– Familiar with GPU architecture and GPU programming
– Experience with designing software architecture, interfaces and abstraction
– Excellent written and oral communication skills

Extra Awesome:
– Background with deep learning fundamentals, frameworks, graph compilers, LLVM, MLIR etc.
– Experience in developing/optimizing deep learning frameworks such as PyTorch , TensorFlow, JAX etc.
– Familiarity with LLM related techniques e.g., langchain, vector database, prompt engineering
– Track record of leading a project and mentoring junior engineers.
– Track record of maintaining or contributing to open source projects.

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
This role is a Position of Trust. Should you accept this position, you must consent to and pass an extended Background Investigation, which includes (subject to country law), extended education, SEC sanctions, and additional criminal and civil checks. For internals, this investigation may or may not be completed prior to starting the position. For additional questions, please contact your Recruiter.

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. * Job posting details (such as work model, location or time type) are subject to change.