Staff Applied Scientist, UI Control Models

What you'll do

• Develop and deploy compact LLMs that provide low-latency, personalized UI control experiences for mobile devices.
• Lead research and engineering efforts in model post-training techniques including supervised fine-tuning and reinforcement learning.
• Implement advanced capabilities for tool calling and complex reasoning within conversational models.
• Optimize model efficiency to ensure high-performance execution on constrained hardware.
• Move AI innovations from the prototype stage through to full production deployment.
• Establish technical directions and lead teams in the end-to-end model lifecycle process.
• Build custom benchmarks and environments to evaluate the performance of UI control agents.

What we're looking for

• Master's or PhD in Computer Science, Machine Learning Engineering, or equivalent professional experience.
• 7+ years of hands-on industry experience shipping ML-powered products or features, ideally involving LLMs.
• Expertise in the ML development loop including model architectures, data curation, training policies, and evaluation.
• Deep expertise in LLM/VLM post-training methods, including supervised fine-tuning and reinforcement learning.
• Proficiency in Python and at least one deep learning framework such as PyTorch, JAX, or TensorFlow.
• Experience developing AI agents for complex reasoning, planning, tool use, and UI control.
• Experience developing small LLMs for constrained or edge devices.
• Experience as a tech lead leading teams of 5 to 10 people.