AI Frameworks Engineer

6 days ago

Bengaluru, Karnataka, India Intel Full time ₹ 12,00,000 - ₹ 36,00,000 per year
Job Details:Job Description: 

We are looking for a dynamic and passionate contributors to work on Intel's 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.
Job Type:Experienced HireShift:Shift 1 (India)Primary Location: India, BangaloreAdditional Locations:Business group:The Software and AI (SAI) Team drives customer value by enabling differentiated experiences through leadership AI technologies and foundational software stacks, products, and services. The group is responsible for developing the holistic strategy for client and data center software in collaboration with OSVs, ISVs, developers, partners and OEMs. The group delivers specialized NPU IP to enable the AI PC and GPU IP to support all of Intel's market segments. The group also has HW and SW engineering experts responsible for delivering IP, SOCs, runtimes, and platforms to support the CPU and GPU/accelerator roadmap, inclusive of integrated and discrete graphics.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 require an on-site presence. * Job posting details (such as work model, location or time type) are subject to change.

  • AI Frameworks Engineer

    6 days ago

    Bengaluru, Karnataka, India Intel Corporation Full time

    Job Details:Job Description:We are looking for a dynamic and passionate contributors to work on Intel's 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...

  • AI Engineer

    6 days ago

    Bengaluru, Karnataka, India Weekday AI Full time ₹ 60,00,000 - ₹ 80,00,000 per year

    This role is for one of the Weekday's clientsSalary range: Rs Rs ie INR 6-8 LPA)Min Experience: 0 yearsLocation: BangaloreJobType: full-timeWe are looking for a passionate and motivated AI Engineer to join our growing team. This role is ideal for individuals who are eager to build a career in artificial intelligence and machine learning. You will work...

  • AI Engineer

    6 days ago

    Bengaluru, Karnataka, India Weekday AI Full time ₹ 6,00,000 - ₹ 8,00,000

    This role is for one of the Weekday's clientsSalary range: Rs Rs ie INR 6-8 LPA)Min Experience: 0 yearsLocation: BangaloreJobType: full-timeWe are looking for a passionate and motivated AI Engineer to join our growing team. This role is ideal for individuals who are eager to build a career in artificial intelligence and machine learning. You will work...

  • AI Engineer

    6 days ago

    Bengaluru, Karnataka, India Weekday AI Full time ₹ 6,00,000 - ₹ 8,00,000 per year

    This role is for one of the Weekday's clientsSalary range: Rs Rs ie INR 6-8 LPA)Min Experience: 0 yearsLocation: BangaloreJobType: full-time We are looking for a passionate and motivated AI Engineer to join our growing team. This role is ideal for individuals who are eager to build a career in artificial intelligence and machine learning. You will work...

  • Agentic Ai Engineer

    8 hours ago

    Bengaluru, Karnataka, India Weekday AI Full time ₹ 5,00,000 - ₹ 8,00,000

    This role is for one of the Weekday's clientsSalary range: Rs Rs ie INR 5-8 LPA)Min Experience: 3 yearsLocation: bangaloreJobType: full-timeAs an Agentic AI Engineer, you will design and build advanced AI-driven voice and conversational systems using LLMs, agentic frameworks, and speech intelligence models. This role involves developing scalable AI agents,...

  • Agentic Ai Engineer

    5 hours ago

    Bengaluru, Karnataka, India Weekday AI Full time ₹ 50,00,000 - ₹ 80,00,000 per year

    This role is for one of the Weekday's clientsSalary range: Rs Rs ie INR 5-8 LPA)Min Experience: 3 yearsLocation: bangaloreJobType: full-timeAs an Agentic AI Engineer, you will design and build advanced AI-driven voice and conversational systems using LLMs, agentic frameworks, and speech intelligence models. This role involves developing scalable AI agents,...

  • Agentic Ai Engineer

    2 hours ago

    Bengaluru, Karnataka, India Weekday AI Full time ₹ 50,00,000 - ₹ 80,00,000 per year

    This role is for one of the Weekday's clientsSalary range: Rs Rs ie INR 5-8 LPA)Min Experience: 3 yearsLocation: bangaloreJobType: full-time As an Agentic AI Engineer, you will design and build advanced AI-driven voice and conversational systems using LLMs, agentic frameworks, and speech intelligence models. This role involves developing scalable AI agents,...

  • AI/ML Engineer

    6 hours ago

    Bengaluru, Karnataka, India Weekday AI Full time ₹ 12,00,000 - ₹ 20,00,000 per year

    This role is for one of the Weekday's clientsSalary range: Rs Rs ie INR 20-35 LPA)Min Experience: 4 yearsLocation: Bengaluru, MysoreJobType: full-timeWe are seeking a dynamic and forward-thinking AI/ML Engineer to contribute to cutting-edge innovation and drive the exploration, prototyping, and adoption of emerging technologies. This role offers an exciting...

  • AI/ML Engineer

    3 hours ago

    Bengaluru, Karnataka, India Weekday AI Full time ₹ 24,00,000 - ₹ 42,00,000 per year

    This role is for one of the Weekday's clientsSalary range: Rs Rs ie INR 20-35 LPA) Min Experience: 4 yearsLocation: Bengaluru, MysoreJobType: full-time We are seeking a dynamic and forward-thinking AI/ML Engineer to contribute to cutting-edge innovation and drive the exploration, prototyping, and adoption of emerging technologies. This role offers an...

  • AI/ML framework Staff Engineer

    6 days ago

    Bengaluru, Karnataka, India Qualcomm Full time ₹ 20,00,000 - ₹ 25,00,000 per year

    General Summary:Looking for ML framework and AI compiler Engineer responsible forDesigning, implementing, and deploying machine learning models using PyTorchFocusing on backend infrastructure and system architecture.Responsibilities often include developing framework, integrating with other AI tools, and ensuring scalability and reliability.Here's a more...