Skip to main content
Posted 05 June, 2026

AI ML Engineer - Lead

Armakuni
Ahmedabad, GJ, IN Full Time
Reference: fd2563f396bb5483

Job Description We are looking for a highly skilled Lead AI/ML Engineer with 6+ years of hands-on experience in designing, deploying, and scaling AI/ML...

Job Description

We are looking for a highly skilled Lead AI/ML Engineer with 6+ years of hands-on experience in designing, deploying, and scaling AI/ML and Generative AI solutions. The ideal candidate will lead technical efforts across a wide spectrum of AI technologies, including large language models (LLMs), computer vision, machine learning, and data science, to solve high-impact business problems.


This is a technical leadership role that involves mentoring team members, shaping architecture decisions, and driving end-to-end AI solution delivery across cloud-based environments.


Key Responsibilities:

Leadership & Strategy

  • Drive the technical vision and roadmap for AI/ML and Generative AI initiatives.
  • Lead and mentor a team of AI/ML engineers and data scientists, conducting regular code reviews and guiding solution architecture.
  • Collaborate with cross-functional stakeholders to translate business problems into scalable AI/ML solutions.
  • Own the AI/ML lifecycle—from prototyping and experimentation to production deployment and monitoring.

Generative AI & LLMs

  • Architect and lead the development of scalable Generative AI solutions using AWS Cloud and modern frameworks.
  • Deep experience in working with large language models (LLMs) including fine-tuning, evaluation, and deployment.
  • Expertise in prompt engineering, agentic workflows, and custom LLM development.
  • Proficient with Langchain, Langgraph, Langfuse, Crew AI, and other GenAI frameworks.
  • Build and optimize Retrieval-Augmented Generation (RAG) pipelines for enterprise applications.

Machine Learning & Data Science

  • Solve complex machine learning problems using supervised, unsupervised, and deep learning methods.
  • Lead the implementation of ML pipelines including automated data preprocessing, training, validation, and monitoring.
  • Manage ML lifecycle using AWS SageMaker, including model versioning, drift detection, and retraining strategies.
  • Apply advanced analytics, time series forecasting, and predictive modeling techniques.

Computer Vision

  • Design and deploy advanced computer vision models for tasks such as classification, object detection, segmentation, and image captioning.


Skills & Qualifications:

  • 6+ years of experience in AI/ML, with at least 3 years in Generative AI and LLMs.
  • Strong proficiency in Python and ML libraries such as Scikit-learn, XGBoost, LightGBM, CatBoost.
  • Deep understanding of LLMs including OpenAI (GPT-4, GPT-4o), Claude, Gemini, LLaMa, DeepSeek, etc.
  • Proficiency with GenAI tooling: Langchain, Langfuse, Crew AI, LLAMA Index, Langgraph, etc.
  • Experience working with vector databases (Pinecone, FAISS, OpenSearch, Chroma) and indexing strategies.
  • Hands-on with deep learning frameworks such as PyTorch and TensorFlow.
  • Strong experience with AWS services: SageMaker, Bedrock, DynamoDB, S3, Lambda, etc.
  • Proficiency in REST API development using FastAPI, Uvicorn, Flask, Docker, and related tools.
  • Strong database knowledge—both SQL (PostgreSQL) and NoSQL (DynamoDB).
  • Familiarity with caching systems like Redis and Memcached.
  • Strong communication skills and ability to lead discussions with both technical and non-technical stakeholders.

Preferred:

  • Experience in leading enterprise-scale AI/ML deployments.
  • Contributions to open-source GenAI/ML projects.
  • Certifications in AWS AI/ML or equivalent.

This listing expired on 16 Jun. Applications are no longer accepted.

Below are some other jobs we think you might be interested in.

  • AI/ML Engineer- Lead
    • Crescens Inc.
    • Bangalore,Karnataka,India
    Job Title: AI/ML Engineer- Lead Location: Bangalore, Karnataka (onsite) Total Experience: 7 to 9 years Mode of Interview: Face to Face (Mandatory)...
    27 May
  • AI/ML Engineer- Lead
    • National Staffing
    • Bangalore,Karnataka,India
    Job Title: AI/ML Engineer- Lead Location: Bangalore, Karnataka (onsite) Total Experience: 7 to 9 years Mode of Interview: Face to Face (Mandatory)...
    27 May
  • AI/ML Engineer- Lead
    • Crescens Inc.
    • Bangalore, Karnataka, IN
    Job Title: AI/ML Engineer- Lead Location: Bangalore, Karnataka (onsite) Total Experience: 7 to 9 years Mode of Interview: Face to Face (Mandatory)...
    26 May
  • AI ML Engineer - Lead
    • Armakuni
    • Ahmedabad, GJ, IN
    Job Description We are looking for a highly skilled Lead AI/ML Engineer with 6+ years of hands-on experience in designing, deploying, and scaling AI/ML...
    16 Jun
  • AI / ML Engineer/Lead
    • Codvo.ai
    • Pune, Maharashtra, IN
    At Codvo, software and people transformations go hand-in-hand. We are a global empathy-led technology services company where product innovation and...
    12 Jun
  • AI / ML Engineer/Lead
    • Codvo.ai
    • Maharashtra,Pune,India
    At Codvo, software and people transformations go hand-in-hand. We are a global empathy-led technology services company where product innovation and...
    12 Jun
  • Lead AI/ML Engineer
    • Fission Labs
    • Hyderabad, TG, IN
    Job Description Role: Lead AI & ML/ Associate Architect -AI/ML\nExperience: 8+ years\nLocation: Hyderabad\nFission Labs , headquartered in Sunnyvale,...
    16 Jun
  • Lead AI/ML Engineer
    • RingCentral
    • Bengaluru, KA, IN
    Job Description Job Description:\nWe are seeking an experienced Lead AI/ML Engineer with a strong background in Natural Language Understanding (NLU)...
    13 Jun
  • Lead AI ML Engineer
    • 3Pillar
    • Noida
    We build breakthrough software products that power digital businesses. We are an innovative product development partner whose solutions drive...
    12 Jun
  • Lead ML & AI Engineer
    • HERE Technologies
    • Mumbai,IN,400063
    What's the role?The HERE NDS Technologies team is looking for a Lead AI/ML Engineer to drive NDS mapping systems. You will lead the design and...
    12 Jun
  • Lead AI/ML
    • Tekfortune India IT Pvt Ltd.
    • Indore, Madhya Pradesh, IN
    What will your role look like Lead the design, development, and deployment of AI and Generative AI solutions using AWS, Azure, or GCP. Provide...
    12 Jun
  • Lead AI/ML
    • TekFortune
    • Indore,Madhya Pradesh,452001
    What will your role look like Lead the design, development, and deployment of AI and Generative AI solutions using AWS, Azure, or GCP. Provide...
    12 Jun
  • AI/ML Lead
    • Open Financial Technologies
    • Bengaluru, KA, IN
    Job Description About Open: OPEN is a leading connected finance platform that empowers finance teams to manage their cash flow better by integrating...
    16 Jun
  • AI/ML Engineer
    • NR Consulting - India
    • Bangalore, Karnataka, IN
    Title: AI/ML Engineer Location: Bangalore Exp: 4 to 6 Years Job Description: " 3 8 years of experience in software development, AI/ML...
    27 May
  • AI/ ML Engineer
    • MaieuticSemi
    • Bangalore Urban, Karnataka, IN
    AI/ ML Engineer Position Overview We are seeking a skilled AI/ML Engineer with expertise in developing and deploying classical machine learning models...
    15 Jun
  • AI/ML Engineer
    • NR Consulting
    • Bangalore,Karnataka
    Title: AI/ML Engineer Location: Bangalore Exp: 4 to 6 Years Job Description: 3-8 years of experience in software development, AI/ML...
    29 May
  • AI/ML Engineer
    • Mindfire Solutions
    • Dombivali, MH, IN
    Job Description About the Job\nAs an AI/ML Engineer, you will be responsible for architecting, developing, and deploying production-grade Generative AI...
    16 Jun
  • AI/ML Engineer
    • Mindfire Solutions
    • Malappuram, KL, IN
    Job Description About the Job As an AI/ML Engineer, you will be responsible for architecting, developing, and deploying production-grade Generative AI...
    13 Jun
  • AI/ML Engineer
    • Mindfire Solutions
    • Panipat, HR, IN
    Job Description About the Job As an AI/ML Engineer, you will be responsible for architecting, developing, and deploying production-grade Generative AI...
    17 Jun
  • AI/ML Engineer
    • Mindfire Solutions
    • Thāne, MH, IN
    Job Description About the Job As an AI/ML Engineer, you will be responsible for architecting, developing, and deploying production-grade Generative AI...
    17 Jun