24 days ago
San Mateo, CA, USAMid Level
H1B sponsor
Base Salary
$173k - $210k/yr
Responsibilities
- Build and support training pipelines and machine learning model implementations to accelerate experimentation.
- Adapt open-source LLM infrastructure such as DeepSpeed and OpenRLHF for InMobi-specific post-training needs.
- Optimize online and batch inference systems 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 improve training and inference capabilities.
- Explore compute platforms and ecosystems such as JAX on TPUs.
- Collaborate with Applied Scientists on experiment design and modeling research.
Requirements
- Master’s degree in machine learning or a related field is required; a PhD is a plus.
- At least 3 years of experience building large-scale machine learning or deep learning systems as a Machine Learning Engineer.
- Expertise in large-scale distributed data systems, relational and key-value stores, Airflow, and Spark.
- Expertise in Python; familiarity with Java, particularly for high-performance serving systems, is a plus.
- Expertise in PyTorch or a similar machine learning ecosystem.
- Experience with accelerator-based inference systems such as Triton and optimization backends including ONNX and TensorRT.
- Experience with cloud platforms and Kubernetes.
- Knowledge of deep recommender systems, reinforcement learning systems, and advertising technologies such as OpenRTB, SSPs, DSPs, ad exchanges, and MMPs is a plus.
Benefits
- Base salary range of $172,500—$210,000 USD for California and New York offices, with regional variation by location.
- Equity in the form of Restricted Stock Units may be available.
- Medical, dental, and vision insurance, including a company-matched HSA.
- 401(k) company match.
- Vacation time, sick days, special occasion time, and company-wide holidays.
- Maternity and paternity leave benefits.
- Flexible working hours.
- Wellness stipend.
- Free lunch in company offices.
- Pet-friendly work environment and pet insurance.
- Employee Assistance Program.
- Equal employment opportunity and reasonable workplace accommodations.
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.