Top Machine Learning Engineer Jobs in Mumbai

The Staff Engineer in Machine Learning will guide teams, solve client challenges, and enhance solutions leveraging their technical expertise.

Lead technology vision and architecture for enterprise projects; design cloud-native, data-driven and Generative AI solutions; ensure scalability, security, governance, and engineering excellence; run architecture reviews, POCs, and implementations; evaluate emerging technologies; influence clients via workshops and executive engagements; mentor architects and distributed engineering teams.

Partner with strategic enterprise customers to define GenAI strategy, design architectures using ChatGPT/OpenAI API, build prototypes to production, manage stakeholder relationships, scale best practices and notebooks, and provide high-signal feedback to Product and Research teams.

Early-career role analyzing PB-scale on-chain, transactional and customer data. Perform data engineering, ETL, querying, build visualizations and ML models, translate insights for teams, manage reporting environment, secure confidential data, and troubleshoot reports/databases.

Analyze PB-scale trading, transactional, operational and customer data; build visualizations, ML models and experiments; translate findings into actionable recommendations for trading, operations, product and executives in a fast-paced environment.

As a Customer Support Engineer, you will resolve customer technical issues, perform installations, upgrade systems, and support field services while maintaining customer relationships.

The Data Operations Analyst develops custom product attribution based on client needs, utilizing tagging and machine learning; ensuring data mapping accuracy and project delivery.

One-year, full-time fellowship supporting design, delivery, and evaluation of interventions for smallholder farmers. Duties include research and analysis, pilot design, field implementation, M&E, and cross-team collaboration with heavy mentorship and rotating in-country placements.

Design, develop, and maintain UI and API automation for AI/ML and generative AI applications. Own end-to-end testing (functional, integration, regression), validate model outputs (accuracy, relevance, safety, hallucinations), build reusable frameworks, integrate tests into CI/CD, collaborate with engineering and product teams, and mentor junior engineers to promote automation-first AI quality practices.

Design and implement scalable end-to-end AI/ML solutions and data pipelines; define secure cloud and on-prem architectures; lead MLOps, CI/CD, model lifecycle management, data governance, and cross-functional collaboration to drive AI initiatives in manufacturing.

Develop and support secure IoT software applications, APIs, and UIs; build dashboards, ETL/data workflows, and analytics; enable ML/AI capabilities; maintain and modernize legacy code hosted on AWS.

Design, train, fine-tune, evaluate, and deploy production machine learning models for cybersecurity detection. Build efficient transformer and LLM systems optimized for latency, throughput, memory, cost, and reliability. Develop scalable inference infrastructure, data pipelines, and labeling workflows; collaborate with security researchers and product teams; monitor production models and improve detection quality and robustness.

Lead an 8-10 engineer team owning courier systems (dispatch, matching, pricing, ETA). Set technical direction, drive data-driven decisions and experiments, partner with data science, ensure reliability and scalability of low-latency distributed systems, and own hiring, performance, and operational excellence.

Lead architecture and development of the core native iOS SDK and related apps. Own feature delivery from design to public release, write high-quality tested code, collaborate with R&D and ML teams, support customers to resolve issues, and manage priorities across large-scale, distributed projects.

Lead post-deployment operations and optimization of Agentic AI, LLM integrations, and Python-based analytical pipelines. Supervise a data science operations team, ensure secure high-availability production systems, manage integrations with enterprise systems (SAP, Salesforce, Postgres), and drive continuous improvement, monitoring, and adoption of analytics solutions.

Architect and improve autonomous AI optimization algorithms for a marketing platform. Analyze A/B tests, integrate AI models, develop multi-touch attribution, expand advertising-channel capabilities, reduce inference costs and latency, incorporate customer feedback, and maintain reliable, continuously improving optimization loops.

Consult with CPG/FMCG clients on Revenue Growth Management analytics programs, focusing on pricing strategies, promotion analysis, and optimizing revenue growth while leading engagements across diverse global markets.

As an AI/ML Engineer, you will develop machine learning models, collaborate with teams, preprocess data, and optimize ML pipelines. Additionally, you will mentor junior members and contribute to AI/ML strategies.

No responsibilities are provided in the job description. The title "Cloud Machine Learning Engineer" implies work on developing, deploying, and maintaining machine learning models and infrastructure in cloud environments.

The Associate Architect - Machine Learning role involves developing cloud ML solutions on AWS, fine-tuning LLMs, and collaborating with teams for optimal model implementation.

Architect and implement enterprise-grade MLOps/LLMOps on AWS (EKS-first), building scalable pipelines for training, validation, registry, deployment, monitoring, CI/CD, observability, lineage and governance; serve as technical authority, conduct architecture reviews, troubleshoot systems, and mentor engineering teams across projects.

This role involves leading the development of AI systems to replace lab toxicity experiments, focusing on end-to-end ML architecture, research on chemistry-biology correlations, and building innovative models. Candidates should have strong ML skills, engineering abilities in Pytorch and Python, and a passion for both science and technology.

Design, develop, deploy, and maintain ML/AI solutions for LMS and student portals. Build recommendation engines, chatbots, predictive analytics, and data pipelines. Integrate models with web apps, support UAT, troubleshooting, documentation, and coordinate with product, tech teams, and vendors to ensure reliable, secure, and scalable ML systems.

The role involves analyzing biological mass spectrometry data, developing software tools, and scaling workflows for AI and human consumption in drug safety assessment.

Lead data science initiatives using advanced statistical techniques and machine learning to produce actionable insights. Partner with cross-functional teams to design data-driven solutions, manage and ensure integrity of large datasets, develop predictive models to forecast trends, and communicate results to stakeholders while staying current on emerging technologies.