Senior AI/ML Engineer

Milestone Technologies, Inc.; Hyderabad; On-site; INR 2,800,000 - 4,000,000; Full time

Job summary

Milestone Technologies, Inc. is seeking an experienced AI Engineer in Hyderabad to implement and test capabilities across feature suites, including RAG pipeline integration and prompt tuning.

You will deliver GraphRAG-based dynamic query generation, graph-aware retrieval, and knowledge-graph schema evolution support. You will collaborate with the AI Architect and Backend Engineer, working on orchestration integration and upgrading the RAG-layer retrieval pipeline to guide agent selection and

Qualifications

• 6+ years AI engineering experience with strong Python fundamentals.
• Hands-on RAG pipeline and prompt-engineering experience.
• LLM application development experience and NLP/semantic engineering background.
• 3+ years RAG/GraphRAG implementation experience and vector store integration.
• Familiarity with SPARQL and graph-aware retrieval.
• Experience building or validating golden Q&A datasets for grounding LLM systems.

Responsibilities

• Develop feature logic for assigned capabilities within assigned feature suites.
• Design and tune prompts and RAG pipeline integrations.
• Validate sprint-level acceptance criteria against the golden dataset baseline.
• Coordinate with the AI Architect and Backend Engineer on design and integration questions.
• Build GraphRAG-based dynamic query generation to replace template-based SPARQL querying.
• Deliver graph-aware retrieval agent cards.
• Support knowledge-graph schema evolution.
• Coordinate with the KG validation workstream on schema and validation questions.
• Deliver full orchestration integration with the platform's cookbook-based orchestration component.
• Integrate the RAG-layer retrieval pipeline to inform agent selection, routing and response generation.
• Document the upgrade path to fine-tuning (documentation only).
• Expand the golden dataset and semantic grounding.
• Maintain question-and-answer baseline categorization.

Skills

Python

RAG pipeline

LLM

Prompt engineering

GraphRAG

Vector store

SPARQL

NLP

Ontology mapping

Semantic engineering

Education

B.Tech/MCA in CS/AI

Tools

GraphRAG

LangGraph

SPARQL

Job description

Role summary

Implements and tests assigned capabilities across feature suites, including RAG pipeline integration and prompt tuning; delivers GraphRAG-based dynamic query generation, graph-aware retrieval, and knowledge-graph schema evolution support; delivers full orchestration integration for the platform's orchestration component.

Key responsibilities

• Develop feature logic for assigned capabilities within assigned feature suites
• Design and tune prompts and RAG pipeline integrations
• Validate sprint-level acceptance criteria against the golden dataset baseline
• Coordinate with the AI Architect and Backend Engineer on design and integration questions
• Build GraphRAG-based dynamic query generation to replace template-based SPARQL querying
• Deliver graph-aware retrieval agent cards
• Support knowledge-graph schema evolution
• Coordinate with the KG validation workstream on schema and validation questions
• Deliver full orchestration integration with the platform's cookbook-based orchestration component
• Integrate the RAG-layer retrieval pipeline to inform agent selection, routing and response generation
• Document the upgrade path to fine-tuning (documentation only, not an implementation, in this phase)
• Expand the golden dataset and semantic grounding
• Maintain question-and-answer baseline categorization

Required skills & experience

• 6+ years AI engineering experience with strong Python fundamentals
• Hands-on RAG pipeline and prompt-engineering experience
• LLM application development experience
• Comfortable validating work against a capped, governed golden dataset rather than open-ended tuning
• 3+ years RAG / GraphRAG implementation experience
• Vector store integration experience
• Familiarity with SPARQL and graph-aware retrieval
• Strong Python skills
• 3+ years RAG implementation and orchestration integration experience
• Vector store / retrieval pipeline integration experience
• Comfortable scoping a fine-tuning upgrade path as a design recommendation rather than an implementation
• 3+ years NLP / semantic engineering experience
• Ontology mapping and semantic disambiguation experience
• Experience building or validating golden Q&A datasets for grounding LLM systems

Preferred qualifications

• Experience on a regulated or enterprise AI product
• Familiarity with LangGraph or comparable agent frameworks
• Direct experience replacing a template-based SPARQL approach with GraphRAG
• Experience coordinating with a separate KG validation team
• Experience with cookbook-style orchestration frameworks
• Experience writing forward-looking technical design documents (vs. shipping code) when scope calls for it
• Experience working within a per-sprint expansion cap rather than open-ended dataset growth
• Pharma/manufacturing terminology disambiguation experience

Education

B.Tech/MCA in Computer Science, AI/ML, or a related field, or equivalent practical experience.