Staff Engineer I

What is the InMobi family like?  

We are an infectious bunch. Be it the way we rise to challenges, the innovative products we create, the dreams we chase or the fun we have at work. We are sure that if you meet us, you will be infected too. Today, we are proud to be the leaders in Mobile advertising and are on an accelerated path of being a leader in enterprise software for marketers. We invite you to free yourself, dream big and chase your passion. We are here today because a few of us did just that.  

What can we promise?  

We offer you an opportunity to work on building enterprise platforms that require you to acquire, hone and demonstrate your ML and AI skills in designing and solving for complex problems, thereby creating software that creates value for our customers in a reliable, scalable, secure, and AI-first manner. At InMobi, you get to work on innovative technologies, work closely with business teams and grow as an overall technology leader. 

Modern work environment, flexible schedule, and smart, creative, down-to-earth people. Internal opportunities to move roles and try out bridge assignments with different teams.  Food for your soul – free meals all days of the week, gym, or yoga class to flex those biceps, cocktails at drink cart Thursdays and fun at work on Funky Fridays. We even promise to let you bring your kids and pets to work. 

What is the team like? 

At InMobi Enterprise team, we are transforming the way finance, sales, marketing, people, and legal functions operate. We’re driving transformative work across Sales, Marketing, People, and Finance with ML and Agentic AI—and we’re looking for passionate innovators to join us!

What you will be doing

We are looking for a Data Scientist who can operate at the intersection of classical machine learning, GenAI and modern agentic AI systems with a strong engineering appetite. You will build and deploy intelligent AI/ML systems that will help transform the business process across the core. This role blends deep ML craftsmanship with forward-looking innovation in autonomous/agentic systems.

Your responsibilities will include:

• Lead the experimentation, design and build of agentic AI systems that autonomously observe model performance, trigger experiments, tune hyperparameters, improve ranking policies, or orchestrate ML workflows with minimal human intervention.
• Apply LLMs, embeddings, retrieval-augmented architectures, and multimodal generative models for semantic understanding, content classification, and user preference modeling.
• Design intelligent agents that can automate repetitive decision-making tasks—e.g., candidate generation tuning, feature selection, or context-aware content curation.
• Support POCs/ Pilots projects to explore and validate innovative technologies and solutions in ML/AI space.

• Build and operate machine learning models on diverse, high-volume data sources for forecasting, classification and prediction.
• Develop rapid experimentation workflows to validate hypotheses and measure real-world business impact.
• Design pipelines for data preparation, model training, evaluation, and deployment in collaboration with engineering counterparts.
• Define best practices for monitoring ML model performance using statistical techniques; identifying drifts, failure modes, and improvement opportunities.
• Contribute to ML/AI thought leadership—blogs, case studies, internal tech talks, and industry conferences.
• Thrive in a multi-functional, highly collaborative team environment with engineering, product, business, and creative teams.
• Interface with stakeholders across Product, Business, Data, and Infrastructure to align ML initiatives with strategic priorities.
• Document designs, development processes, and best practices to promote knowledge sharing and operational efficiency.

What do we expect from you?

We are seeking candidates with deep expertise in ML and hands-on experience building GenAI and Agentic AI systems.

Skillset: AI/ML Engineering, Python, LLM, RAG, Agentic AI, MCP, REST API

You should have experience with:

• Design, build, and deploy building LLM-powered applications, chatbots, and multi-agent systems to improve efficiency, accuracy, and decision-making.
• Implement and customize GenAI and multi-agent systems using AI frameworks such as LangChain, LangGraph, CrewAI, or similar.
• Create, refine, and maintain prompt templates, chains, and workflows.
• Integrate vector databases (MilvusDB, FAISS, Pinecone, Chroma, etc.) for semantic search and memory.
• Classical ML and deep learning techniques across NLP, clustering, and time series.
• Proficient in deploying ML workflows/models in production system. Continuous monitoring, evaluation, and retraining of models to ensure sustained performance. Ensuring model explainability, accuracy, compliance.
• Big data processing (Spark, distributed data systems) and cloud computing.
• Designing end-to-end ML solutions—from prototype to production.
• Stay updated with emerging AI/ML technologies and recommend relevant innovations for the finance domain.
• Drive in design reviews, sprint planning, and technical discussions.
• Collaborating effectively with diverse stakeholders, including technical teams and business leaders.
• Adapting to fast-paced environments and evolving priorities with high energy and autonomy.

We value curiosity, problem-solving ability, and a strong bias toward experimentation and production impact.

 Qualifications

• Bachelor’s/Master’s in Computer Science, Statistics, Mathematics, Electrical Engineering, Operations Research, Economics, Analytics, or related fields. PhD is a plus.
• 8+ years of industry experience in ML/Data Science, with deep proficiency in Python and one or more frameworks: PyTorch, TensorFlow, Scikit-learn.
• Solid understanding of ML fundamentals, regression, tree-based models, clustering, and time series.
• Hands-on experience with LLMs, retrieval systems, generative models, or agentic/autonomous ML systems is highly desirable.
• Experience building Agentic frameworks such as LangGraph, AutoGen, CrewAI, n8n or ReACT-style agents.
• Expertise with algorithms in NLP, Time Series, and Deep Learning, applied on real-world datasets.
• Strong experience with the big data ecosystem (Spark, Hadoop) and cloud platforms (Azure, AWS, GCP/Vertex AI).
• Experience with Model Context Protocols (MCPs) — building or integrating MCP tools, servers, or capabilities.
• Comfortable working in cross-functional teams.
• Strong problem-solving and analytical skills with the ability to work in agile environments.
• Good understanding of data pipelines, APIs, and cloud platforms (Azure/GCP/AWS).
• Excellent communication skills with the ability to simplify complex technical concepts.