Lead design, development, and deployment of production-grade ML models, AI agents, and serverless data pipelines on AWS. Architect ML workflows, MLOps/CI-CD, data ingestion and model-serving infrastructure, integrate external systems, mentor engineers, and translate business needs into scalable AI solutions.
Senior AI/ML Engineer
About the Role
We're seeking a Senior AI/ML Engineer to lead the design, development, and deployment of ML/AI solutions, agents, and data automations across our AWS-based analytics platform. In this role, you'll architect end-to-end ML systems, set technical direction, mentor junior engineers and analysts, and drive best practices in MLOps and ML infrastructure. This is a high-impact role for an experienced engineer who can own complex, ambiguous problems and deliver production-grade machine learning and AI agent solutions.
Key Responsibilities
- Architect, train, evaluate, and optimize machine learning models, owning the full model lifecycle from experimentation to production
- Design and build AI agents and automated workflows using Amazon Quick and AWS orchestration tools
- Define and implement efficient ML workflows, optimizing for performance, scalability, and cost
- Architect and maintain serverless data pipelines using AWS Glue, Step Functions, Lambda, and EventBridge Scheduler
- Lead the design of our analytics service engine for ingesting, transforming, and querying data across S3 storage (Excel/CSV files, Delta Tables, library files)
- Establish MLOps practices, CI/CD pipelines, and infrastructure standards for the team
- Integrate with external systems (e.g., SAP, Salesforce) and design robust data-sourcing strategies
- Design and build REST APIs and model-serving infrastructure for production workloads
- Mentor junior engineers, conduct code reviews, and set technical standards
- Partner with cross-functional stakeholders to translate business needs into ML solutions
Required Technical Skills
- Programming: Expert in Python with a track record of writing clean, well-tested, production-grade code
- ML Frameworks: Strong, hands-on experience with PyTorch, TensorFlow, and scikit-learn
- Model Development: Deep understanding of model training, evaluation, inference, and optimization for efficient ML at scale
- AI Agents & Automation: Proven experience building AI agents and automated workflows; proficient with Amazon Quick
- MCP & Tool Integration: Experience building and integrating Model Context Protocol (MCP) servers to connect LLMs and AI agents with external tools, data sources, and services
- APIs & Serving: Strong experience designing REST APIs and deploying/serving ML models in production
- Cloud & Infrastructure: Solid experience with cloud platforms (AWS, Azure, GCP) and containerization (Docker)
- MLOps & Tooling: Proficient with Git, CI/CD pipelines, and ML infrastructure best practices
Familiarity with Our Architecture
Our team's analytics platform is built on AWS. Deep familiarity with the following components is expected, and you'll help shape how we use and evolve them:
- Orchestration & Compute: AWS Glue, AWS Step Functions, AWS Lambda, EventBridge Scheduler
- Storage & Data: S3 (CSV/Excel, Delta Tables, library files), Glue Data Catalog, Glue Crawler
- Query & Analytics: Amazon Athena
- AI & Automation: Amazon Quick
- Integration: External systems such as SAP and Salesforce
- Notifications: Amazon SNS and Amazon SES for alerting and email
- Infrastructure & Security: AWS IAM, AWS Secrets Manager, CloudWatch, AWS Systems Manager (for environment parameters)\
- Source Control & CI/CD: Bitbucket for version control, pull request workflows, and pipeline-based deployments
Core Competencies
- Problem-Solving: Independently solves complex, ambiguous ML and automation challenges and designs scalable solutions
- Technical Leadership: Sets technical direction, drives architecture decisions, and mentors junior engineers
- Collaboration: Leads cross-functional initiatives and owns the delivery of significant components end-to-end
- Continuous Improvement: Champions MLOps, software engineering best practices, and ML infrastructure across the team
- Code Quality: Sets and enforces high standards through clean, testable code, rigorous reviews, and robust version control
Similar Jobs
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
Lead design, development, training, evaluation, and productionization of scalable ML/AI models using MLOps and CI/CD. Collaborate with data, product, and engineering teams, mentor engineers, enforce ML best practices, and ensure responsible, ethical AI and model monitoring in production.
Top Skills:
AWSAzureCi/CdComputer VisionDeep LearningDockerGCPKubernetesMlopsModel MonitoringNlpPythonPyTorchScikit-LearnTensorFlow
Artificial Intelligence • Big Data • Cloud • Information Technology • Machine Learning • Software
Lead design, development, and operation of production AI/ML systems (LLMs, RAG, multi-agent). Drive architectures, MLOps, model evaluation, cloud deployments, and mentor junior engineers to align AI with product needs.
Top Skills:
AWSCi/CdEmbeddingsExperiment TrackingLlmMlMlopsMulti-Agent SystemsNlpPythonRagReinforcement LearningRetrieval SystemsSemantic Search
Healthtech • Software
Design and develop machine learning infrastructure and models, support product teams, build scalable data services, and promote collaboration within a diverse engineering team.
Top Skills:
AWSDagsterDvcGCPJupyterKubeflowMlflowPostgresPythonTriton Server
What you need to know about the Pune Tech Scene
Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.
