AI Frameworks Engineer

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 senior software developers to join us and optimize open source AI frameworks, libraries and models on Intel CPUs/GPUs/accelerators. In this role, you will profile and analyze AI and deep learning workloads and get them optimized on Intel hardware by contributing to open source frameworks and libraries such as PyTorch, JAX, HuggingFace, vLLM, Deepspeed etc to benefit developers and customers in an out of box fashion.

Qualifications:You are good at:
- Master, PhD or equivalent in relevant fields (Computer Science, EE, Math)
- 2+ years of relevant working experience
- Programming skills in C, C++, and Python
- Familiar with GPU architecture and GPU programming
- Background in deep learning and neural networks
- Good communication and teamwork
Extra Awesome:
- Experience with deep learning frameworks, graph compilers, LLVM, MLIR etc.
- Experience in performance analysis and optimization.
- Familiarity with LLM related techniques e.g., langchain, vector database, prompt engineering
- Track record of 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 TrustN/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. * Job posting details (such as work model, location or time type) are subject to change.