4 hours ago
San Mateo, CA, USAMid Level
H1B sponsor
Base Salary
$173k - $210k/yr
Responsibilities
- Build and support training pipelines and machine learning model implementations.
- Adapt open-source LLM infrastructure such as 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 for infrastructure reliability and model behavior.
- Use product data features to strengthen data pipelines and enable training and inference capabilities.
- Explore platforms and ecosystems such as JAX on TPUs to diversify compute.
- Collaborate with Applied Scientists on experiment design and modeling.
- Develop and optimize ad science models and the infrastructure supporting them at massive scale.
Requirements
- Master’s degree in machine learning or a related field is required; a PhD is a plus.
- At least 3 years of experience as a Machine Learning Engineer building state-of-the-art machine learning or deep learning systems at extreme scale.
- Expertise in large-scale distributed data systems, high-performance relational and key-value stores, Airflow, and Spark data transformations.
- Expertise in Python; familiarity with Java, particularly for 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 such as ONNX and TensorRT.
- Experience with cloud platforms and Kubernetes-based container orchestration.
- Knowledge of deep recommender systems, including two-tower and attention-based architectures, is a plus.
- Knowledge of reinforcement learning systems, reward signals, and post-training feedback loops is a plus.
- Understanding of the advertising ecosystem, including OpenRTB, SSPs, DSPs, ad exchanges, and MMPs, is a plus.
Benefits
- Base salary ranges from $172,500 to $210,000 USD for offices in California and New York, with equity in the form of RSUs where applicable.
- Medical, dental, and vision insurance, including a company-matched HSA, are provided.
- 401(k) company match is provided.
- Benefits include vacation time, sick days, special occasion time, and company-wide holidays.
- Substantial maternity and paternity leave benefits are available.
- Flexible working hours are offered.
- The role includes a wellness stipend, free daily lunch in offices, pet insurance, and an Employee Assistance Program.
- The company provides continuous learning and career progression through the InMobi Live Your Potential program.
- The posting applies to offices in California and New York and notes that salary ranges vary by final location and region.
Tech Stack
Categories
About InMobi
InMobi builds a mobile advertising and consumer engagement platform for advertisers and publishers and operates Glance, a lock‑screen content and commerce service. It sells programmatic ad buying, monetization tools, and AI-driven personalization across apps, TVs, and the open web. Founded in 2007 and headquartered in Singapore, its ad network reaches over 2 billion people in 150+ countries and serves more than 30,000 brands.