IN-Manager _AI Architect __ GCC_ Advisory_ Bangalore

PwC; Bengaluru; On-site; INR 4,000,000 - 6,000,000; Full time

Job summary

PwC Bengaluru invites an AI Architect to design robust ML pipelines, APIs for model inference, and scalable LLM serving. The role demands deep knowledge of GPU architectures, distributed training, and modern MLOps with Kubernetes and Docker.

You will work with platforms like SageMaker, Vertex AI, and Azure AI to deliver data-driven business insights. Ideal candidates have 3+ years in AI/ML architecture, strong Python and ML framework skills, and a product-focused mindset to solve complex client

Qualifications

• Experience designing ML pipelines for experiment management, model management and retraining.
• Proven ability to design APIs for scalable model inferencing.
• Strong knowledge of ML platforms like SageMaker, Vertex AI and Azure AI.

Responsibilities

• Design ML pipelines for experiment management, model management, feature management and retraining.
• Design APIs for scalable model inferencing at scale.
• Demonstrate expertise in GPU architectures and distributed training/serving of LLMs.
• Lead DevOps and LLMOps practices with Kubernetes, Docker and related tooling.

Skills

Gen AI

LLM

HuggingFace

Python

PyTorch

TensorFlow

Keras

LangChain

LangGraph

Docker

Kubernetes

ML Ops

Education

BE/B.Tech

MBA

MCA

Tools

SageMaker

Vertex AI

Azure AI

LangFlow

Flowise

LlamaIndex

Kubernetes

Docker

Job description

Line of ServiceAdvisoryIndustry/SectorNot ApplicableSpecialismData, Analytics & AIManagement LevelManagerJob Description & SummaryAt PwC, our people in data and analytics focus on leveraging data to drive insights and make informed business decisions. They utilise advanced analytics techniques to help clients optimise their operations and achieve their strategic goals.In data analysis at PwC, you will focus on utilising advanced analytical techniques to extract insights from large datasets and drive data-driven decision-making. You will leverage skills in data manipulation, visualisation, and statistical modelling to support clients in solving complex business problems.*Why PWCAt PwC, you will be part of a vibrant community of solvers that leads with trust and creates distinctive outcomes for our clients and communities. This purpose-led and values-driven work, powered by technology in an environment that drives innovation, will enable you to make a tangible impact in the real world. We reward your contributions, support your wellbeing, and offer inclusive benefits, flexibility programmes and mentorship that will help you thrive in work and life. Together, we grow, learn, care, collaborate, and create a future of infinite experiences for each other. Learn more about us .At PwC, we believe in providing equal employment opportunities, without any discrimination on the grounds of gender, ethnic background, age, disability, marital status, sexual orientation, pregnancy, gender identity or expression, religion or other beliefs, perceived differences and status protected by law. We strive to create an environment where each one of our people can bring their true selves and contribute to their personal growth and the firm’s growth. To enable this, we have zero tolerance for any discrimination and harassment based on the above considerations. \"Job Description & Summary: AI ArchitectResponsibilities:ML Pipeline Design:Design ML pipelines for experiment management, model management, feature management, and model retraining.Design APIs for model inferencing at scale.Proven expertise with MLflow, SageMaker, Vertex AI, and Azure AI.LLM Serving and GPU Architecture:Possess deep knowledge of GPU architectures.Expertise in distributed training and serving of large language models.Proficient in model and data parallel training using frameworks like DeepSpeed and service frameworks like vLLM.Model Fine-Tuning and Optimization:Demonstrate proven expertise in model fine-tuning and optimization techniques.Achieve better latencies and accuracies in model results.Reduce training and resource requirements for fine-tuning LLM and LVM models.DevOps and LLMOps Proficiency:Proven expertise in DevOps and LLMOps practices.Knowledgeable in Kubernetes, Docker, and container orchestration.Deep understanding of LLM orchestration frameworks like Flowise, Langflow, and Langgraph.Skill MatrixLLM: Hugging Face OSS LLMs, GPT, Gemini, Claude, Mixtral, LlamaLLM Ops: ML Flow, Langchain, Langraph, LangFlow, Flowise, LLamaIndex, SageMaker, AWS Bedrock, Vertex AI, Azure AIDatabases/Datawarehouse: DynamoDB, Cosmos, MongoDB, RDS, MySQL, PostGreSQL, Aurora, Spanner, Google BigQuery.Cloud Knowledge: AWS/Azure/GCPDev Ops (Knowledge): Kubernetes, Docker, FluentD, Kibana, Grafana, PrometheusCloud Certifications (Bonus): AWS Professional Solution Architect, AWS Machine Learning Specialty, Azure Solutions Architect ExpertProficient in Python, SQL, JavascriptMandatory skill sets:Gen AI,LLM, Huggingface, python,pytorch/tensor flow/keras, Langchain, Langgraph, Docker, KunernetesPreferred skill sets:SQL,machine learning, data scienceYears of experience required:3+Education qualification:BE/B.Tech/MBA/MCAEducation (if blank, degree and/or field of study not specified)Degrees/Field of Study required: Master of Business AdministrationDegrees/Field of Study preferred:Certifications (if blank, certifications not specified)Required SkillsJavaScript Object Notation (JSON)Optional SkillsAccepting Feedback, Accepting Feedback, Active Listening, AI Fluency, AI-Human Collaboration, Algorithm Development, Alteryx (Automation Platform), Analytical Thinking, Analytic Research, Big Data, Business Data Analytics, Coaching and Feedback, Communication, Complex Data Analysis, Conducting Research, Creativity, Customer Analysis, Customer Needs Analysis, Dashboard Creation, Data Analysis, Data Analysis Software, Data Collection, Data-Driven Insights, Data Integration, Data Integrity {+ 46 more}Desired Languages (If blank, desired languages not specified)Travel RequirementsNot SpecifiedAvailable for Work Visa Sponsorship?NoGovernment Clearance Required?NoJob Posting End DateOctober 27, 2026