Machine Learning Engineer – Ad Sciences

Key details

Full job description

InMobi Advertising seeks a Machine Learning Engineer to develop and optimize ad science models, open-source LLMs, and infrastructure focusing on performance and cost. The role involves building training pipelines, adapting LLM infrastructure (DeepSpeed, OpenRLHF), optimizing inference, monitoring systems, leveraging data features, exploring new compute platforms (JAX on TPUs), and collaborating with Applied Scientists. Requirements include a Master's degree in machine learning or related field (PhD a plus), 3+ years experience building large-scale ML/DL systems, expertise in Python, PyTorch, distributed data systems (Airflow, Spark), cloud platforms, Kubernetes, and familiarity with Java, Triton, ONNX, TensorRT. Knowledge of recommender systems, reinforcement learning, and ads ecosystem is a plus. The position is full-time, on-site in San Mateo, CA, with a salary range of $172,500 to $210,000 USD plus benefits including RSUs, medical/dental/vision insurance, 401(k) match, generous leave, flexible hours, wellness stipend, free lunch, pet-friendly environment, and Employee Assistance Program.