Machine Learning Engineer – Ad Sciences

InMobi Advertising is a global technology leader helping marketers win the moments that matter. Our advertising platform reaches over 2 billion people across 150+ countries and turns real-time context into business outcomes, delivering results grounded in privacy-first principles. Trusted by 30,000+ brands and leading publishers, InMobi is where intelligence, creativity, and accountability converge. By combining lock screens, apps, TVs, and the open web with AI and machine learning, we deliver receptive attention, precise personalization, and measurable impact.

Through Glance AI, we are shaping AI Commerce, reimagining the future of e-commerce with inspiration-led discovery and shopping. Designed to seamlessly integrate into everyday consumer technology, Glance AI transforms every screen into a gateway for instant, personal, and joyful discovery.  Spanning diverse categories such as fashion, beauty, travel, accessories, home décor, pets, and beyond, Glance AI delivers deeply personalized shopping experiences. With rich first-party data and unparalleled consumer access, it harnesses InMobi’s global scale, insights, and targeting capabilities to create high impact, performance driven shopping journeys for brands worldwide.

Recognized as a Great Place to Work, and by MIT Technology Review, Fast Company’s Top 10 Innovators, and more, InMobi is a workplace where bold ideas create global impact. Backed by investors including SoftBank, Kleiner Perkins, and Sherpalo Ventures, InMobi has offices across San Mateo, New York, London, Singapore, Tokyo, Seoul, Jakarta, Bengaluru and beyond.

At InMobi Advertising, you’ll have the opportunity to shape how billions of users connect with content, commerce, and brands worldwide. To learn more, visit www.inmobi.com

Overview of the Role:

We are looking for a Machine Learning Engineer to develop and optimize our ad science models, open-source LLMs, and the underlying infrastructure that powers them — with a constant eye toward performance and cost.

AI and ML sit at the core of InMobi's business, and our science teams are tackling some of the field's most interesting problems: creative optimization, user personalization, targeting, yield management, traffic shaping, and more. We're moving fast and deep into architectural research, generative AI adoption, compute scaling on the latest accelerators (GPUs/TPUs), and advances in reinforcement learning.

You'll join a team of experienced researchers and engineers who have been pushing the boundaries of ad science research for over a decade, working closely with data and platform engineering to build solutions that scale and serve the needs of every science team across the organization. This is a chance to shape the infrastructure and models behind ad intelligence operating at truly massive scale.

The Impact You'll Make:

• Build and support training pipelines and model implementations that maximize experimentation velocity
• Adapt open-source LLM infrastructure like DeepSpeed and OpenRLHF to meet InMobi-specific post-training needs
• Optimize online and batch inference for low latency and cost efficiency
• Build monitoring and evaluation solutions that keep our infrastructure reliable and our models behaving as expected
• Leverage the breadth of data features across our product portfolio to strengthen our data pipelines and unlock new features for training and inference
• Explore new platforms and ecosystems (e.g., JAX on TPUs) to help diversify our compute
• Collaborate closely with Applied Scientists on active research, contributing directly to experiment design and modeling

The Experience We Need:

• Master's degree in machine learning or a related field required; a PhD is a plus
• At least 3 years of experience as a Machine Learning Engineer, building state-of-the-art machine learning/deep learning systems at extreme scale
• Expertise in large-scale distributed data systems, including high-performance relational and key-value stores, orchestrators like Airflow, and data transformations in Spark
• Expertise in Python; familiarity with Java (especially in high-performance serving systems) is a plus
• Expertise in PyTorch or a similar ecosystem
• Experience with accelerator-based inference systems such as Triton, and optimization backends like ONNX and TensorRT
• Experience with cloud platforms and container orchestration using Kubernetes
• Knowledge of deep recommender systems, including two-tower and attention-based architectures, is a plus
• Knowledge of reinforcement learning systems — including RL approaches, reward signals, and post-training feedback loops — is a plus
• Understanding of the ads ecosystem (OpenRTB, SSPs, DSPs, ad exchanges, MMPs) is a plus

At InMobi, you’ll be surrounded by people who…

• Think big and act fast: We’re entrepreneurial, thrive in ambiguity, and love solving high-impact problems
• Are passionate, fanatically driven, and take immense pride in their work: We care deeply about the impact we create and continuously push our potential
• Own their outcomes: We take responsibility, make bold decisions, and execute with confidence
• Embrace freedom with accountability: We value autonomy and understand that trust comes with responsibility
• Believe in lifelong learning: We welcome feedback, challenge ourselves to grow, and aren’t afraid to take smart risks

Award-winning culture, best-in-class benefits

Our compensation philosophy enables us to provide a competitive salary that drives high performance while balancing business needs and pay parity. We determine compensation based on a wide variety of factors, including role, nature of experience, skills, and location. 

The base (fixed) salary for this role ranges from $172,500—$210,000 USD (min to max of base salary pay range). This salary range applies to our offices in California and New York*.

*Our ranges may vary based on the final location and region of the role in accordance with the geographical differentiation in pay scales in the country.

In addition to cash compensation, based on the position, an InMobian can receive equity in the form of a stock grant known as Restricted Stock Units (RSUs). To encourage a spirit of shared ownership, we believe that our employees should have the ability to own a part of the organization. Furthermore, as you contribute to the growth of the company, additional stock may be issued in recognition of your contribution over time.

A quick snapshot of our benefits:

• Competitive salary and RSU grant (where applicable)        
• High-quality medical, dental, and vision insurance (including company-matched HSA)       
• 401(k) company match 
• Generous combination of vacation time, sick days, special occasion time, and company-wide holidays   
• Substantial maternity and paternity leave benefits and a compassionate work environment 
• Flexible working hours to suit everyone 
• Wellness stipend for a healthier you!      
• Free lunch provided in our offices daily   
• Pet-friendly work environment and robust pet insurance policy - because we love our animals! 
• Employee Assistance Program (EAP)

If this sounds like the kind of team, role, values, benefits, and challenges that excite you, we’d love to hear from you. Go for it!

InMobi is an equal opportunity employer      

InMobi is a place where everyone can grow. Howsoever you identify, and whatever background you bring with you, we invite you to apply if this sounds like a role that would make you excited to work.  

InMobi provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type.  All qualified applicants will receive consideration for employment without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state, or local laws. This policy applies to all terms and conditions of employment, including recruiting, hiring, placement, promotion, termination, layoff, recall, transfer, leaves of absence, compensation, and training. 

The InMobi Culture

At InMobi, culture isn’t a buzzword; it's an ethos woven by every InMobian, reflecting our diverse backgrounds and experiences.

We thrive on challenges and seize every opportunity for growth. Our core values — thinking big, being passionate, showing accountability, and taking ownership with freedom — guide us in every decision we make.

We believe in nurturing and investing in your development through continuous learning and career progression with our InMobi Live Your Potential program.

InMobi is proud to be an Equal Employment Opportunity employer and is committed to providing reasonable accommodations to qualified individuals with disabilities throughout the hiring process and in the workplace.

Visit https://www.inmobi.com/company/careers to better understand our benefits, values, and more!