Engineering Specialist - GenAI

Zs Associates; Pune District; Hybrid; INR 1,800,000 - 2,400,000; Full time

Benefits offered by this job

Hybrid work model

Travel opportunities

Job summary

ZS Associates is seeking an Architecture & Engineering Specialist – ML Engineering in India to design and deploy scalable ML models. The role collaborates with data scientists and client teams to build robust ML pipelines using Sagemaker, Kubeflow, and MLFlow, while ensuring code quality and maintainability.

You will thrive in a fast-paced environment with travel for client engagements. You should have 5+ years in ML model deployment, strong Python/Java/Scala skills, and expertise in Hive,

Qualifications

• Minimum 5+ years of hands-on experience deploying ML models at scale.
• Experience with ML engineering platforms like Sagemaker, Kubeflow or MLFlow.
• Strong knowledge of big data technologies (Hive, Spark, Hadoop).
• Ability to run POCs on new technologies and document findings.

Responsibilities

• Design and implement technical features leveraging current tech stack.
• 2
• Collaborate with client-facing teams to understand solution context and gather requirements.
• Write production-ready code that is testable and handles edge cases.
• Ensure high-quality deliverables by following standards and code reviews.
• Write unit and integration tests to cover happy and error paths.
• Use bug tracking, version control, and other tools to deliver work.
• Participate in scrum calls and agile ceremonies.
• Continuously evaluate new technologies and create prototypes.
• Support the project architect in module design and integration.
• Break features into estimable tasks and defend them with clients.
• Debug complex issues using logs, traces, and monitoring tools.
• Mentor junior engineers and grow technical talent.

Skills

ML Engineering

Python

PySpark

Java/Scala

Spark

Hive

Kafka

Sagemaker

Kubeflow

MLFlow

Tools

AWS

GCP

Kubernetes

Airflow

Job description

ZS is a place where passion changes lives. As a management consulting and technology firm focused on improving life and how we live it, we transform ideas into impact by bringing together data, science, technology and human ingenuity to deliver better outcomes for all. Here you’ll work side-by-side with a powerful collective of thinkers and experts shaping life-changing solutions for patients, caregivers and consumers, worldwide. ZSers drive impact by bringing a client-first mentality to each and every engagement. We partner collaboratively with our clients to develop custom solutions and technology products that create value and deliver company results across critical areas of their business. Bring your curiosity for learning, bold ideas, courage and passion to drive life-changing impact to ZS.

Architecture & Engineering Specialist – ML Engineering

ZS’s India Capability & Expertise Center (CEC)

houses more than 60% of ZS people across three offices in New Delhi, Pune and Bengaluru. Our teams work with colleagues across North America, Europe and East Asia to create and deliver real world solutions to the clients who drive our business. The CEC maintains standards of analytical, operational and technological excellence across our capability groups. Together, our collective knowledge enables each ZS team to deliver superior results to our clients.

ZS's Scaled AI practice

is part of ZS's rich and advanced AI ecosystem, in the Architecture & Engineering Expertise Center, focused on creating continuous business value for clients using a range of innovative machine learning, deep learning, and engineering capabilities.

Being part of Scaled AI practice allows you to collaborate with data scientists to create state-of-the-art AI models, create and use cutting-edge ML platforms, create and deploy advanced ML pipelines and manage the complete ML lifecycle.

What You'll Do

• Design and implement technical features leveraging best practices for technology stack being used
• Collaborate with client-facing teams to understand solution context and contribute to technical requirement gathering and analysis
• Work with technical architects on the team to validate design and implementation approach
• Write production-ready code that is easily testable, understood by other developers, and accounts for edge cases and errors
• Ensure the highest quality of deliverables by following architecture/design guidelines, coding best practices, periodic design/code reviews
• Write unit tests as well as higher-level tests to handle expected edge cases and errors gracefully, as well as happy paths
• Uses bug tracking, code review, version control, and other tools to organize and deliver work
• Participate in scrum calls and agile ceremonies, and effectively communicate work progress, issues, and dependencies
• Consistently contribute in researching & evaluating the latest technologies through rapid learning, conducting proofs-of-concept and creating prototype solutions
• Support the project architect in designing modules/component of the overall project/product architecture
• Breaks down large features into estimable tasks lead estimation and can defend them with clients
• Implement complex features with limited guidance from the engineering lead. For example service or application-wide change
• Systematically debug code issues/bugs using stack traces, logs, monitoring tools, and other resources
• Performs code/script reviews of senior engineers in the team
• Mentor and groom technical talent within the team

What You'll Bring

• At least 5+ relevant hands-on experience in deploying and productionizing ML models at scale
• Expertise in Designing, configuring, and using ML Engineering platforms like Sagemaker, MLFlow, Kubeflow, or other platforms
• Big data - Hive, Spark, Hadoop, queuing system like Apache Kafka/Rabbit MQ/AWS Kinesis
• Ability to quickly adapt to new technology and be innovative in creating solutions
• Ability to independently run POCs on new technologies and document findings to share
• Strong in at least one of the Programming languages - PySpark, Python or Java, Scala, etc. and Programming basics - Data Structures
• Hands-on experience in building metadata-driven, reusable design patterns for data pipeline, orchestration, ingestion patterns (batch, real-time)
• Fluency in English
• Client-first mentality
• Intense work ethic
• Collaborative spirit and problem-solving approach

How you'll grow

• Cross-functional skills development & custom learning pathways
• Milestone training programs aligned to career progression opportunities
• Internal mobility paths that empower growth via s-curves, individual contribution and role expansions

Perks & Benefits

At ZS, your growth matters. We offer a comprehensive total rewards package that supports your health and well‑being, financial future, time away, and professional development. With robust skills‑building programs, multiple career progression paths, internal mobility, and a deeply collaborative culture, you’ll have the opportunity to do meaningful work, expand your capabilities, and thrive as part of a global community. For details on total rewards in India, visit ZS India office locations | Where we work | ZS.

Hybrid working model

We are committed to giving our employees a flexible and connected way of working. A flexible and connected ZS allows us to combine work from home and on-site presence at clients/ZS offices for the majority of our week. The magic of ZS culture and innovation thrives in both planned and spontaneous face-to-face connections.

Travel

Travel is a requirement at ZS for client facing ZSers; business needs of your project and client are the priority. While some projects may be local, all client-facing ZSers should be prepared to travel as needed. Travel provides opportunities to strengthen client relationships, gain diverse experiences, and enhance professional growth by working in different environments and cultures.

ZS is an equal opportunity employer and is committed to providing equal employment and advancement opportunities without regard to any class protected by applicable law.