New opportunity

Lead AI/ML Engineer

ARMAKUNI INDIA · Ahmedabad, India

About this role

ARMAKUNI INDIA seeks a full-time Lead AI/ML Engineer in Ahmedabad to lead the design, deployment and scaling of AI/ML and Generative AI solutions. The role combines hands-on engineering, technical leadership and end-to-end delivery across cloud environments.

Responsibilities

  • Set the technical vision and roadmap for AI/ML and Generative AI initiatives. Mentor AI/ML engineers and data scientists, review code, guide architecture and work with stakeholders to turn business problems into scalable solutions.
  • Own the lifecycle from prototyping and experimentation through production deployment and monitoring. Architect Generative AI solutions on AWS; fine-tune, evaluate and deploy LLMs; develop prompts, agentic workflows and custom LLM solutions; and build and optimize enterprise RAG pipelines.
  • Apply supervised, unsupervised and deep learning methods. Lead automated data preprocessing, training, validation and monitoring pipelines; manage model versioning, drift detection and retraining with SageMaker; and use advanced analytics, time series forecasting and predictive modeling.
  • Design and deploy computer vision models for classification, object detection, segmentation and image captioning.

Requirements

  • At least 6 years of AI/ML experience, including at least 3 years in Generative AI and LLMs. Strong Python skills and experience with Scikit-learn, XGBoost, LightGBM or CatBoost; PyTorch or TensorFlow; and LLMs such as GPT-4, GPT-4o, Claude, Gemini, LLaMa and DeepSeek.
  • Proficiency with GenAI tools including LangChain, Langfuse, CrewAI, LlamaIndex and LangGraph; vector databases such as Pinecone, FAISS, OpenSearch and Chroma; and indexing strategies.
  • Strong experience with AWS SageMaker, Bedrock, DynamoDB, S3 and Lambda; REST API development using FastAPI, Uvicorn, Flask and Docker; SQL and NoSQL databases, including PostgreSQL and DynamoDB; and caching systems such as Redis and Memcached. Strong communication and the ability to lead technical and non-technical discussions are required.

Preferred

Experience leading enterprise-scale AI/ML deployments, open-source GenAI/ML contributions, and AWS AI/ML or equivalent certifications.

For full details about this role, click “Apply on company site” to view the employer’s original posting.

Skills for this role

PythonMachine LearningGenerative AILarge Language ModelsPrompt EngineeringAI AgentsRAGComputer VisionDeep LearningSupervised LearningUnsupervised LearningTime Series ForecastingPredictive ModelingScikit-learnXGBoostLightGBMCatBoostPyTorchTensorFlowLangChainLangGraphLangfuseCrewAILlamaIndexPineconeFAISSOpenSearchChromaAWSAmazon SageMaker Plus API v2.1.0 via Keka SSR Connect API 1.2.0 page payload 229

Your skill match

Checking your profile…

YOUR NEXT STEP

Get interview-ready for this role

A focused preparation guide, built around this job’s responsibilities and requirements.

✦ AI-generated guide
Preparation suggestions, not the employer’s actual interview questions. Always check the original posting for current requirements.
01

Understand the role

What matters most

  • Set direction and lead delivery

    The role owns the AI/ML and Generative AI roadmap, architecture guidance, code reviews and mentoring. It also calls for translating business problems with technical and non-technical stakeholders into scalable solutions.

  • Build and evaluate enterprise GenAI systems

    The posting emphasizes AWS-based Generative AI, LLM fine-tuning, evaluation and deployment, prompt engineering, agentic workflows, and optimized RAG pipelines.

  • Own production ML lifecycles

    Responsibility extends from experiments to deployment and monitoring, including automated preprocessing, training and validation, plus SageMaker model versioning, drift detection and retraining strategies.

  • Deliver across ML and computer vision

    Beyond LLMs, the role covers supervised, unsupervised and deep learning; forecasting and predictive modeling; and deployed vision models for classification, detection, segmentation and image captioning.

  • Connect models to cloud applications

    The requirements call for AWS services including SageMaker and Bedrock, REST API development, SQL and NoSQL databases, and caching—not just model development.

Likely challenges to think through

  • Choosing the right GenAI approach

    A likely difficulty is deciding when prompting, RAG, agentic workflows or fine-tuning warrants the added complexity while maintaining useful evaluation. Prepare to explain decision criteria and how you would compare approaches; the posting does not specify a current project or preferred choice.

  • Moving experiments into maintainable production

    Rapid prototyping can conflict with the effort needed for automated pipelines, versioning, monitoring and retraining. Prepare an example or design walkthrough that makes those handoffs and trade-offs explicit, without assuming ARMAKUNI INDIA has a particular deployment issue.

  • Balancing technical depth with leadership breadth

    The remit spans LLMs, conventional ML, computer vision, cloud integration and team guidance, making prioritization a likely challenge. Prepare to explain how you would scope work, delegate reviews and align architecture decisions with stakeholder needs; these are preparation hypotheses, not confirmed team problems.

03

Practice your answers

Technical practice

How would you turn a business problem into an AI/ML solution roadmap?

What to coverThe posting calls for technical vision and translating stakeholder problems into scalable solutions. Suggested preparation: explain how you would define success measures, compare approaches, and sequence a prototype, evaluation, and production rollout.

How would you design and evaluate an enterprise RAG pipeline?

What to coverThe posting names RAG, vector databases, and indexing strategies. Suggested preparation: discuss document preparation, retrieval and indexing choices, answer-quality evaluation, and how you would diagnose poor retrieval.

When would you fine-tune an LLM rather than improve prompts or retrieval?

What to coverThe posting covers LLM fine-tuning, evaluation, deployment, and prompt engineering. Suggested preparation: compare the approaches using task-specific evidence, evaluation results, operational complexity, and cost.

How would you design and troubleshoot an agentic LLM workflow?

What to coverThe posting calls for agentic workflows and names frameworks including LangGraph, Crew AI, and Langfuse. Suggested preparation: explain how you would control workflow steps, inspect failures, and evaluate whether an agent adds value over a simpler approach.

How would you choose among classical ML and deep learning approaches?

What to coverThe posting names supervised, unsupervised, and deep learning methods, alongside Scikit-learn, boosting libraries, PyTorch, and TensorFlow. Suggested preparation: compare data needs, baselines, validation methods, interpretability, and deployment constraints for a defined problem.

How would you build and maintain an automated ML pipeline in SageMaker?

What to coverThe posting specifies preprocessing, training, validation, monitoring, model versioning, drift detection, and retraining. Suggested preparation: describe how you would connect those stages and distinguish a meaningful drift signal from one that does not warrant retraining.

How would you validate a time-series forecasting model?

What to coverThe posting includes time-series forecasting and predictive modeling. Suggested preparation: discuss time-aware validation, appropriate baselines and error measures, and how you would investigate changing performance after deployment.

How would you select and evaluate a computer vision approach?

What to coverThe posting covers classification, object detection, segmentation, and image captioning. Suggested preparation: explain how the task determines data labeling, model choice, evaluation measures, and deployment trade-offs.

How would you divide responsibilities among the named AWS services for an AI solution?

What to coverThe posting names SageMaker, Bedrock, S3, DynamoDB, and Lambda for cloud-based delivery. Suggested preparation: outline a possible architecture, justify each service's role, and explain how you would assess scaling and operational trade-offs.

How would you serve an AI model through a production REST API?

What to coverThe posting names FastAPI, Uvicorn, Flask, Docker, PostgreSQL, DynamoDB, Redis, and Memcached. Suggested preparation: discuss API design, deployment, persistence and caching choices, and how you would diagnose slow or unreliable responses.

Behavioural practice

Tell me about a time you mentored an engineer or improved work through code review.

What to coverThe posting calls for mentoring and regular code reviews. Suggested preparation: use a real situation, describe the feedback and follow-through you provided, and state the resulting change without overstating your contribution.

Describe a time you aligned technical and non-technical stakeholders on an AI solution.

What to coverThe posting emphasizes cross-functional collaboration and leading discussions with both audiences. Suggested preparation: explain the business problem, how you communicated options and trade-offs, and what decision or outcome followed.

Tell me about a time you set technical priorities for an AI/ML initiative.

What to coverThe posting assigns ownership of technical vision and roadmap. Suggested preparation: describe the situation, how you weighed competing needs, the actions you took to set priorities, and the result.

Describe an AI/ML project you took from experimentation toward production.

What to coverThe posting calls for end-to-end ownership through deployment and monitoring. Suggested preparation: explain the initial goal, the decisions and obstacles during delivery, and what you observed after release.

Tell me about an architecture disagreement you helped resolve.

What to coverThe posting calls for guiding architecture and leading technical discussions. Suggested preparation: give the situation, explain how you evaluated alternatives and involved others, and describe the decision and its outcome.

02

Sharpen your knowledge

Topics to prepare

  • Technical leadership and solution architecture

    The role owns the AI/ML roadmap, architecture decisions, code reviews and mentoring. Rehearse how you would turn a business problem into a scoped solution, explain trade-offs to technical and non-technical stakeholders, and review an implementation plan.

  • LLM adaptation and evaluation

    The posting calls for LLM fine-tuning, evaluation, deployment and prompt engineering. Revise when prompting versus fine-tuning is appropriate, then sketch an evaluation plan with baseline results, test cases and failure analysis.

  • Enterprise RAG and agentic design

    RAG pipelines, vector databases, indexing strategies and agentic workflows are explicit responsibilities. Compare two indexing approaches and trace how retrieval quality, prompt design and workflow steps affect an answer.

  • Production ML lifecycle

    The role covers experimentation through monitoring, including SageMaker model versioning, drift detection and retraining. Draw a lifecycle diagram and identify the evidence you would check before deploying or retraining a model.

  • Predictive ML and forecasting

    The posting specifies supervised, unsupervised and deep learning, plus time series forecasting. Revise suitable validation and evaluation choices for each; explain how a forecasting split avoids using future observations.

  • Computer vision task design

    Classification, object detection, segmentation and image captioning are named. For each task, identify the required output and an appropriate way to evaluate it, then explain what would change at deployment.

Skills to revise

  • Python model comparison

    As practice for the stated Python and ML-library requirements, build a small reproducible comparison using Scikit-learn and one of XGBoost, LightGBM or CatBoost; document preprocessing, validation and results.

  • RAG workflow implementation

    As a preparation exercise, prototype retrieval and answer generation using a named framework such as LangChain or LlamaIndex and a named index such as FAISS or Chroma. Inspect retrieved passages and test how indexing changes results.

  • AWS lifecycle walkthrough

    The posting names SageMaker, Bedrock, S3 and Lambda. Practice explaining or diagramming a prototype-to-deployment workflow that includes model versions, monitoring and a retraining decision; distinguish demonstrated experience from a proposed design.

  • Vision model experiment

    Using PyTorch or TensorFlow, practice outlining a small classification or detection experiment, including data preparation, validation and inference. This targets the posting’s hands-on deep learning and computer vision work.

  • API and data-serving design

    Sketch a REST inference endpoint using the posting’s FastAPI or Flask options, with Docker packaging. Explain where PostgreSQL or DynamoDB would fit and when Redis or Memcached caching would help; this is a design exercise, not a stated project.

  • Architecture review communication

    Practice a short code-review explanation and a non-technical account of an AI solution’s trade-offs. Both exercises relate directly to the role’s mentoring, architecture guidance and stakeholder discussions.

04

Your preparation checklist

  • Build a dated evidence sheet

    List actual AI/ML and Generative AI/LLM work with dates, your responsibilities, Python tools and measurable outcomes. Use it to discuss the posting’s 6+ years of AI/ML and 3+ years of GenAI/LLM experience without overstating experience.

  • Rehearse a technical leadership walkthrough

    Prepare one real example covering the business problem, technical roadmap, architecture decisions, code reviews or mentoring, stakeholder discussions and production outcome. Distinguish your decisions from the team’s work.

  • Practice an enterprise RAG design exercise

    Sketch a hypothetical AWS-based solution covering document ingestion, indexing and vector-store choice, retrieval, prompting or agent workflow, LLM evaluation and monitoring. Explain when you would consider fine-tuning instead; label the design as practice, not a past deployment.

  • Prepare an ML lifecycle case study

    Walk through a real supervised, unsupervised or forecasting project from preprocessing and validation to deployment and monitoring. Be ready to explain model versioning, drift detection and retraining, including SageMaker where you have used it.

  • Prepare a computer vision example

    If you have relevant work, outline a classification, detection, segmentation or image-captioning project with its data, model choice, evaluation and deployment constraints. Otherwise, prepare a clearly hypothetical design discussion.

  • Diagram a production integration

    Practice explaining how an AI service could expose a REST API, run in a Docker-based deployment and connect to appropriate AWS storage, compute, database or caching services named in the posting. State trade-offs rather than claiming ARMAKUNI INDIA uses that architecture.

  • Draft questions for the employer

    Prepare questions about which AI/ML and GenAI problems take priority, how the team measures production success, what infrastructure is already in place and the lead’s scope for architecture, mentoring and roadmap decisions.

AI-generated from the job information · Prepared Sep 30, 2026 (UTC)Useful preparation, not a guarantee of eligibility or an offer.

Availability last checked: 5 Nov 2026. Status can change after a check; confirm on the employer’s website. Report an issue · Our checking policy

KEEP EXPLORING

Similar AI jobs

Related skills and specializations in India. Matched to this role, not your profile.

Explore more jobs
Finding similar opportunities…