Senior AI/ML Engineer

Milestone Technologies, Inc.; Hyderabad; On-site; INR 3,500,000 - 5,500,000; Full time

Job summary

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

The role focuses on GraphRAG-based dynamic query generation, knowledge-graph schema evolution, and orchestration integration for the platform. You will collaborate with the AI Architect and Backend Engineer to design efficient retrieval, routing, and response strategies, while expanding and validating golden Q&A

Qualifications

• 6+ years AI engineering experience with Python fundamentals.
• 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 / retrieval pipeline 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 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

Prompt engineering

LLM application development

NLP

Semantic engineering

GraphRAG

Vector store integration

SPARQL

Ontology mapping

Education

B.Tech/MCA in CS/AI/ML

Tools

LangGraph

SPARQL

Vector store

Job description

Company Overview

Milestone Technologies is a global IT managed services firm that partners with organizations to scale their technology, infrastructure and services to drive specific business outcomes such as digital transformation, innovation, and operational agility. Milestone is focused on building an employee-first, performance-based culture and for over 25 years, we have a demonstrated history of supporting category-defining enterprise clients that are growing ahead of the market. The company specializes in providing solutions across Application Services and Consulting, Digital Product Engineering, Digital Workplace Services, Private Cloud Services, AI/Automation, and ServiceNow. Milestone culture is built to provide a collaborative, inclusive environment that supports employees and empowers them to reach their full potential. Our seasoned professionals deliver services based on Milestone’s best practices and service delivery framework. By leveraging our vast knowledge base to execute initiatives, we deliver both short-term and long-term value to our clients and apply continuous service improvement to deliver transformational benefits to IT. With Intelligent Automation, Milestone helps businesses further accelerate their IT transformation. The result is a sharper focus on business objectives and a dramatic improvement in employee productivity. Through our key technology partnerships and our people-first approach, Milestone continues to deliver industry-leading innovation to our clients. With more than 3,000 employees serving over 200 companies worldwide, we are following our mission of revolutionizing the way IT is deployed.

Job Overview

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.

• 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.

Compensation

Estimated Pay Range: Exact compensation and offers of employment are dependent on circumstances of each case and will be determined based on job-related knowledge, skills, experience, licenses or certifications, and location.

Our Commitment to Diversity & Inclusion

At Milestone we strive to create a workplace that reflects the communities we serve and work with, where we all feel empowered to bring our full, authentic selves to work. We know creating a diverse and inclusive culture that champions equity and belonging is not only the right thing to do for our employees but is also critical to our continued success.

Milestone Technologies provides equal employment opportunity for all applicants and employees. All qualified applicants will receive consideration for employment and will not be discriminated against on the basis of race, color, religion, gender, gender identity, marital status, age, disability, veteran status, sexual orientation, national origin, or any other category protected by applicable federal and state law, or local ordinance. Milestone also makes reasonable accommodations for disabled applicants and employees.

We welcome the unique background, culture, experiences, knowledge, innovation, self-expression and perspectives you can bring to our global community. Our recruitment team is looking forward to meeting you.