AI Architect

Milestone Technologies, Inc.; Ernakulam; On-site; INR 3,000,000 - 6,000,000; Full time

Job summary

Milestone Technologies, Inc. is seeking an expert AI/ML Solution Architect in India to lead end-to-end AI/ML design, including RAG pipelines, context engineering, memory management, governance, security posture, prompts, and agent workflows.

You will collaborate with the Knowledge Graph Architect, review designs, and own the MCP Integration Framework and GraphRAG strategy to ensure engineering alignment and robust evaluation.

Qualifications

• 8+ years AI/ML solution architecture experience, including RAG pipeline and prompt-framework design.
• Hands-on experience designing multi-agent workflows.
• Strong grasp of platform architecture alignment and compliance / engineering standards.
• Prior experience owning solution architecture for an AI product suite.
• Hands-on experience with AWS and with MCP or comparable agent-integration protocols.
• Experience designing evaluation architecture for AI systems.
• Comfortable verifying engineering-standard compliance alignment jointly with a Solution Architect.

Responsibilities

• Own RAG pipeline patterns, Context Engineering, Memory Management, governance framework, security posture, prompt frameworks, and agent workflow design, Gen AI Architecture.
• Ensure feature designs align with the overall platform architecture and engineering standard
• Partner with the Knowledge Graph Architect on KG-dependent capability design
• Review and approve designs proposed by the AI/ML Engineers before sprint commitment
• Own orchestrator design and the MCP Integration Framework architecture
• Define the GraphRAG query-generation approach for the platform
• Own evaluation architecture for platform AI components.
• Verify engineering-standard compliance of platform designs jointly with the Technical Architect

Skills

AI/ML architecture

RAG pipeline

Context Engineering

Memory Management

Governance framework

Security posture

Prompt frameworks

Agent workflows

AWS

MCP integration

Evaluation architecture

Solution Architecture

Education

B.Tech/M.Tech in Computer Science, AI/ML

Tools

AWS

MCP protocol

MLflow

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.

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

Leads end-to-end AI/ML solution architecture, including RAG pipeline patterns, Context Engineering, Memory Management, governance framework, security posture, prompt frameworks, and agent workflow design, Gen AI Architecture; and owns orchestrator design, the MCP Integration Framework architecture, the GraphRAG query-generation approach, and evaluation architecture.

Key responsibilities

• Own RAG pipeline patterns, Context Engineering, Memory Management, governance framework, security posture, prompt frameworks, and agent workflow design, Gen AI Architecture.
• Ensure feature designs align with the overall platform architecture and engineering standard
• Partner with the Knowledge Graph Architect on KG-dependent capability design
• Review and approve designs proposed by the AI/ML Engineers before sprint commitment
• Own orchestrator design and the MCP Integration Framework architecture
• Define the GraphRAG query-generation approach for the platform
• Own evaluation architecture for platform AI components.
• Verify engineering-standard compliance of platform designs jointly with the Technical Architect

Required Skills & Experience

• 8+ years AI/ML solution architecture experience, including RAG pipeline and prompt-framework design
• Hands-on experience designing multi-agent workflows
• Strong grasp of platform architecture alignment and compliance / engineering standards
• Prior experience owning solution architecture for an AI product suite
• Hands-on experience with AWS and with MCP or comparable agent-integration protocols
• Experience designing evaluation architecture for AI systems
• Comfortable verifying engineering-standard compliance alignment jointly with a Solution Architect

Preferred Qualifications

• Experience architecting across six or more concurrent capability suites
• Familiarity with knowledge-graph-backed retrieval
• Experience defining a formal MCP registration / agent-card standard
• Evaluation architecture experience tied to MLflow or similar

Education

B.Tech/M.Tech 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 ouremployees butis 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.