Skip to main content
Posted 16 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 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.

Sign up for Job Alerts