AI/ML framework Staff Engineer

3 days ago


Bengaluru, Karnataka, India Qualcomm Full time ₹ 12,00,000 - ₹ 36,00,000 per year

General Summary:

Looking for ML framework and AI compiler Engineer responsible for

  • Designing, implementing, and deploying machine learning models using PyTorch
  • Focusing 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 detailed breakdown of what you might see in such a job description:

Key Responsibilities:

  • Model Development and Deployment: Designing, building, and deploying AI models, particularly those leveraging PyTorch for deep learning.
  • Backend Infrastructure: Developing and maintaining the backend systems that power AI applications, including data ingestion, processing, and storage.
  • System Architecture: Designing scalable and high-performance backend architectures to handle AI workloads.
  • Model Optimization: Optimizing model performance for speed, accuracy, and resource efficiency.
  • Integration: Integrating AI models with other systems and applications.
  • API Development: Creating and maintaining APIs for communication between frontend and backend components.
  • Data Handling: Managing data ingestion, preprocessing, and storage for AI training and inference.
  • Collaboration: Working with data scientists, product managers, and other engineers to bring AI solutions to life.

Tools, Technologies, Skills and Programming:

  • C, C++: Strong programming capability using advanced techniques to design and develop AI compilers and backends.
  • Scripting: Strong expertise in Python with design, develop, release and maintain projects.
  • AI Frameworks:Familiarity with other AI frameworks like PyTorch, TensorFlow, Hugging Face, etc.

  • Machine Learning Knowledge:Understanding of machine learning principles and algorithms starting Computer vision to large language models and continuously update to new trends.

  • Expertise to deep learning accelerator programming (GPU, NPU). Any parallel programming experience (Like CUDA, OpenCL, MKLDNN ..etc) is a plus.
  • Experience with deep leaning compilers like Glow, TVM etc is a plus.

Minimum Qualifications:

Bachelor's degree in Engineering, Information Systems, Computer Science, or related field and 4+ years of Systems Engineering or related work experience. ORMaster's degree in Engineering, Information Systems, Computer Science, or related field and 3+ years of Systems Engineering or related work experience. ORPhD in Engineering, Information Systems, Computer Science, or related field and 2+ years of Systems Engineering or related work experience.


  • AI /Ml Engineer

    4 weeks ago


    Bengaluru, Karnataka, India Weekday AI Full time

    This role is for one of Weekday's clientsSalary range: Rs 1200000 - Rs 2000000 (ie INR 12-20 LPA)Min Experience: 6 yearsLocation: BengaluruJobType: full-timeRequirementsAbout the role- Design, develop, and implement AI/ML models and algorithms.- Build Proof of Concept (POC) applications to demonstrate feasibility and business value.- Write clean, efficient,...

  • AI /Ml Engineer

    5 days ago


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

    This role is for one of Weekday's clientsSalary range: Rs Rs ie INR 12-20 LPA)Min Experience: 6 yearsLocation: BengaluruJobType: full-timeAbout the roleDesign, develop, and implement AI/ML models and algorithms. Build Proof of Concept (POC) applications to demonstrate feasibility and business value. Write clean, efficient, and well-documented code....

  • AI /Ml Engineer

    5 days ago


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

    This role is for one of Weekday's clientsSalary range: Rs Rs ie INR 12-20 LPA)Min Experience: 6 yearsLocation: BengaluruJobType: full-timeRequirementsAbout the roleDesign, develop, and implement AI/ML models and algorithms.Build Proof of Concept (POC) applications to demonstrate feasibility and business value.Write clean, efficient, and well-documented...


  • Bengaluru, Karnataka, India Contrails AI Full time ₹ 15,00,000 - ₹ 25,00,000 per year

    Company DescriptionWe are an early stage VC-backed AI startup based in SF & Bangalore, working in the online safety space - building solutions for next generation problems like deepfake & synthetic media investigation, hate speech, multimodal content moderation, AI guardrailing and more.Role DescriptionThis is a full-time opportunity for an ML/AI Applied...

  • AI/ML Engineer

    3 days ago


    Bengaluru, Karnataka, India Howdee AI Full time ₹ 60,000 - ₹ 6,00,000 per year

    We're looking for a passionateAI/ML engineerto join us as part of the founding team and grow into a future CTO role.This is not just another job , it's a chance to build core technology from scratch, shape the product vision with the founder, and create impact at scale. If you've ever wanted to be part of a lean, ambitious startup at the ground floor, this...

  • ML Engineer

    1 week ago


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

    This role is for one of Weekday's clientsMin Experience: 3 yearsLocation: KarnatakaJobType: full-timeWe are seeking a skilled and innovative Machine Learning (ML) Engineer with a minimum of 3 years of experience in designing, building, and deploying ML models and solutions. The ideal candidate will have strong expertise in machine learning algorithms, data...

  • Ai Ml Engineer

    1 day ago


    Bengaluru, Karnataka, India Agiliad Full time ₹ 5,00,000 - ₹ 25,00,000 per year

    We are seeking a highly skilled AI/ML Engineer with expertise in Python/Java to design, develop, and deploy scalable AI and machine learning solutions. The ideal candidate will have strong problem-solving skills, hands-on experience with ML frameworks, and the ability to work closely with cross-functional teams to translate business problems into data-driven...

  • AI/ML Engineer

    5 days ago


    Bengaluru, Karnataka, India TalentBridge Full time US$ 90,000 - US$ 1,20,000 per year

    Role: AI/ML Engineer (Contract-to-Hire)Experience: 4–5 yearsWhat we are looking for:Hands-on AI/ML project experience (must-have).Strong in Python programming.Exposure to Apache Spark and/or Databricks for large-scale data processing.Experience with Microsoft Azure (preferably Azure ML / AI services).Familiarity with modern AI frameworks like ...

  • ML Engineer

    1 week ago


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

    This role is for one of Weekday's clientsMin Experience: 3 yearsLocation: KarnatakaJobType: full-timeRequirementsWe are seeking a skilled and innovative Machine Learning (ML) Engineer with a minimum of 3 years of experience in designing, building, and deploying ML models and solutions. The ideal candidate will have strong expertise in machine learning...

  • ML Engineer

    1 week ago


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

    This role is for one of Weekday's clientsMin Experience: 3 yearsLocation: KarnatakaJobType: full-timeRequirementsWe are seeking a skilled and innovative Machine Learning (ML) Engineer with a minimum of 3 years of experience in designing, building, and deploying ML models and solutions. The ideal candidate will have strong expertise in machine learning...