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 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

    8 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

    5 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...


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

    CompanyQualcomm India Private LimitedJob AreaEngineering Group, Engineering Group > Systems EngineeringGeneral SummaryLooking 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...


  • 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...


  • Bengaluru, Karnataka, India AsicBot AI Full time ₹ 20,00,000 - ₹ 40,00,000 per year

    About UsWe're revolutionizing semiconductor design with AI. Every smartphone, car, and AI system depends on chips that currently take years and hundreds of millions to design. Our AI-powered platform automates complex design tasks, dramatically reducing time-to-market and development costs in the $600B+ semiconductor industry.Currently bootstrapped and in...

  • ML Engineer

    2 weeks 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

    6 days ago


    Bengaluru, Karnataka, India Volto It Labs Full time ₹ 8,00,000 - ₹ 12,00,000 per year

    The role involves hands-on work in Machine Learning Engineering and Modelling, with strong proficiency in Python programming, GPU-based frameworks such as TensorFlow, Keras, and PyTorch, and Cloud-based ML development and deployment on AWS or Azure.Key Responsibilities:Design, develop, and deploy scalable ML models and solutions.Work with GPU-based...

  • ML Engineer

    2 weeks 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...

  • Ai/ML Engineer

    1 week ago


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

    • Minimum 5 years of experience in machine learning engineering and AI development• Deep expertise in machine learning algorithms and techniques like supervised/unsupervised learning, deep learning, reinforcement learning, etc.• Solid experience in natural language processing (NLP) - language models, text generation, sentiment analysis, etc.• Proven...

  • AI/ ML Engineer

    2 weeks ago


    Bengaluru, Karnataka, India ELBN SYS HR Full time ₹ 20,00,000 - ₹ 25,00,000 per year

    We are supporting a fast-growing AI-driven Start Up that is building the next generation of enterprise AI agents to transform workflows at scale. If you are passionate about Large Language Models, Retrieval-Augmented Generation (RAG), and multi-agent systems, this is your chance to shape the future of enterprise automation.Responsibilities•Design and...