AI/ML Architect

Job Title: AI/ML Architect

Location: Bangalore / Pune

Experience: 10+ Years

Job Type: Full-Time | Permanent

Work Mode: Hybrid

Job Summary

We are seeking an experienced and visionary AI/ML Architect to lead the end-to-end design, development, deployment, and operationalization of advanced AI/ML and Generative AI (GenAI) solutions on cloud platforms. The ideal candidate will possess deep technical expertise in ML architecture, GenAI frameworks, Retrieval-Augmented Generation (RAG) pipelines, cloud-native deployment, and MLOps practices. You will work closely with cross-functional teams, clients, and engineering teams to define scalable AI strategies and deliver cutting-edge solutions across various domains.

Key Responsibilities

Customer Engagement & Solution Architecture

- Interact with clients and stakeholders to gather business and technical requirements and translate them into scalable AI/ML solutions.
- Architect and design AI/ML systems across AWS, GCP, or Azure with a strong focus on cloud-native and cost-optimized architecture.
- Create detailed system design documents, architecture diagrams, and technical roadmaps.
- Define data architecture, storage, and retrieval strategies tailored to AI/ML workflows.

GenAI & RAG Architecture

- Lead the design and implementation of Generative AI solutions using LLMs, LangChain, LlamaIndex, Prompt Engineering, and vector databases such as Pinecone, FAISS, Weaviate, or Elasticsearch.
- Architect RAG (Retrieval-Augmented Generation) pipelines for enterprise use cases including knowledge management, chatbot development, and document summarization.
- Implement prompt orchestration, retrieval optimization, and grounding techniques to enhance LLM output accuracy and relevance.

AI/ML Model Development & MLOps

- Guide the development of Python-based APIs, data preprocessing workflows, and model training pipelines.
- Design and implement robust CI/CD pipelines for ML model deployment using tools like SageMaker, Vertex AI, or Azure ML.
- Define and implement model monitoring, retraining, and performance management strategies for production-grade ML systems.
- Ensure best practices in versioning, reproducibility, model lineage, and auditability (MLOps/LLMOps).

Technical Leadership & Governance

- Review and approve system designs, PoCs, and implementation approaches.
- Provide hands-on leadership and mentorship to data scientists, ML engineers, and software developers.
- Lead architectural decision-making, code quality reviews, and sprint grooming sessions.
- Champion best practices in security, compliance, scalability, and performance optimization for AI/ML solutions.

Project Management & Collaboration

- Own end-to-end technical delivery of AI/ML and GenAI projects across multiple domains (e.g., BFSI, Retail, Healthcare, Manufacturing).
- Coordinate with product owners, business analysts, data engineers, and DevOps teams to ensure seamless delivery.
- Manage stakeholder expectations, project timelines, and resource allocation efficiently.

Required Qualifications

- 10+ years of overall IT experience, with minimum of 5+ years in designing, developing, deploying, and operationalizing AI/ML solutions.
- Minimum 2–3 years of experience in architecting end-to-end AI/ML solutions, including design, implementation, and production deployment.
- Proven experience in GenAI, LLMs, RAG architecture, prompt engineering, and orchestration tools like LangChain, LlamaIndex, etc.
- Hands-on with vector databases (e.g., Pinecone, FAISS, Elasticsearch) and unstructured data retrieval.
- Deep knowledge of Machine Learning and Deep Learning algorithms: CNNs, RNNs, LSTMs, Transformers, etc.
- Experience in Natural Language Processing (NLP), including language modeling, summarization, classification, and NER.
- Strong expertise in Python, with frameworks like PyTorch, TensorFlow, HuggingFace, NumPy, and Pandas.
- Demonstrated experience in designing cloud-native AI/ML solutions on AWS, GCP, or Azure.
- Skilled in deploying models via services like SageMaker, Vertex AI, Azure ML, or using containers and Kubernetes.
- Solid understanding of MLOps/LLMOps lifecycle: pipeline automation, model registry, monitoring, CI/CD.
- Excellent communication, leadership, and stakeholder management skills.

Preferred Qualifications

- Certification in AWS/GCP or ML specializations.
- Experience in leading large-scale AI transformation programs.

Why Join Us?

- Work with cutting-edge GenAI and AI/ML technologies and projects.
- Collaborate with top-tier clients and drive real-world impact.
- Leadership opportunities in a growing AI/ML practice.