Machine Learning Engineer – Ad Sciences
InMobi · San Mateo, CA · $173k–$210k
The role
InMobi Advertising seeks a Machine Learning Engineer to develop and optimize ad science models, open-source LLMs, and the underlying infrastructure, focusing on performance and cost. The role involves building training pipelines, adapting LLM infrastructure, optimizing inference, and collaborating with applied scientists on research. The position requires a Master's degree and at least 3 years of experience in building large-scale ML systems.
- Pay
- $173k–$210k
- Location
- San Mateo, CA
- Level
- Mid
- Experience
- 3+ yrs
- Education
- Masters
What we know that the posting doesn’t say
- Seen todaystill listed on the employer’s careers page
- Posted 7 days agothe first time we saw it
About InMobi
InMobi Advertising is a global technology leader providing an advertising platform that reaches over 2 billion people across 150+ countries, using AI and machine learning for personalized marketing. It also develops Glance AI for AI Commerce, offering personalized shopping experiences across various categories.
What you would do
- Build and support training pipelines and model implementations to maximize experimentation velocity
- Adapt open-source LLM infrastructure like DeepSpeed and OpenRLHF for InMobi-specific post-training needs
- Optimize online and batch inference for low latency and cost efficiency
- Build monitoring and evaluation solutions to ensure infrastructure reliability and model behavior
- Leverage data features across product portfolio to strengthen data pipelines and unlock new features
- Explore new platforms and ecosystems like JAX on TPUs to diversify compute
Must have
- Master's degree in machine learning or related field
- At least 3 years of experience as a Machine Learning Engineer
- Experience building state-of-the-art machine learning and deep learning systems at extreme scale
- Expertise in large-scale distributed data systems including high-performance relational and key-value stores
- Expertise with orchestrators like Airflow and data transformations in Spark
- Expertise in Python
- Expertise in PyTorch or a similar ecosystem
- Experience with accelerator-based inference systems such as Triton
- Experience with optimization backends like ONNX and TensorRT
- Experience with cloud platforms and container orchestration using Kubernetes
Nice to have
- PhD in machine learning or related field
- Familiarity with Java, especially in high-performance serving systems
- Knowledge of deep recommender systems including two-tower and attention-based architectures
- Knowledge of reinforcement learning systems including RL approaches, reward signals, and post-training feedback loops
- Understanding of the ads ecosystem including OpenRTB, SSPs, DSPs, ad exchanges, and MMPs
What you get
- Competitive salary and RSU grant where applicable
- High-quality medical, dental, and vision insurance including company-matched HSA
- 401(k) company match
- Generous vacation, sick days, special occasion time, and company-wide holidays
- Substantial maternity and paternity leave benefits
- Flexible working hours, wellness stipend, free daily lunch, pet-friendly environment, and Employee Assistance Program
Experience and education
- 3+ years of relevant experience
- Masters degree or equivalent experience
Key skills
- python
- pytorch
- spark
- airflow
- kubernetes
- triton
- onnx
- tensorrt
- deepspeed
- openrlhf
- jax
- tpu
- gpu
- relational databases
- key-value stores
- distributed systems
- machine learning
- deep learning
- reinforcement learning
- recommender systems
- llm
- inference optimization
- cloud platforms
- container orchestration
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…
Extracted from the employer’s posting. Read it in full on InMobi’s careers pageApply without filling the form
Approve this role and the application is completed for you, including a résumé tailored to it. You get a confirmation when it lands, and a credit is only spent when a submission is confirmed.
Other roles at InMobi
- Applied Scientist IIISan Mateo, CA
- Applied Scientist – User IntelligenceSan Mateo, CA