Senior AI/ML Engineer

Milestone Technologies, Inc.; Ernakulam; On-site; INR 1,200,000 - 2,400,000; Full time

Job summary

Milestone Technologies, Inc. is seeking an experienced AI engineer to implement and test features across GraphRAG-based pipelines, including prompt tuning and knowledge-graph integration.

You will contribute to RAG-enabled dynamic query generation and the platform's orchestration components. The role requires 6+ years of Python AI engineering, hands-on RAG work, vector store experience, and expertise with SPARQL and graph-aware retrieval.

Qualifications

• 6+ years AI engineering experience with Python
• Hands-on RAG pipeline and prompt-engineering experience
• LLM application development experience
• Comfortable validating work against a capped, governed golden dataset
• 3+ years RAG / GraphRAG implementation experience
• Vector store integration experience
• Familiarity with SPARQL and graph-aware retrieval
• 3+ years NLP / semantic engineering experience
• Ontology mapping and semantic disambiguation experience
• 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, not an implementation, in this phase)
• Expand the golden dataset and semantic grounding
• Maintain question‑and‑answer baseline categorization

Skills

Python

RAG pipeline

Prompt engineering

LLM application development

Vector store

SPARQL

Knowledge graphs

NLP/semantic engineering

Ontology mapping

Golden Q&A datasets

Education

B.Tech/MCA in CS/AI/ML

Tools

Python

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.