AI Research Engineer (Agentic Post-training) - 100% Remote Worldwide

About the role

As a member of the AI model team, you will drive innovation in post-training methodologies, with a special focus on agentic behaviors and tool use. Your work will refine pre-trained models so that they not only deliver enhanced intelligence and domain specific capabilities, but also learn to reason, plan, and autonomously invoke external tools to solve real world, multi step tasks and applications on edge devices (i.e., smartphones).

What you'll do

• Conduct end-to-end research and engineering initiatives to advance post-training of agentic and tool-use models to achieve SOTA results
• Drive broad, cross-cutting model improvements, including factuality, instruction adherence, tool/function use, multi-agent coordination, and reasoning calibration
• Design and enhance large-scale post-training systems, including data pipelines, training workflows, evaluation frameworks, and benchmark infrastructure
• Develop rigorous evaluation suites and diagnostic tools to assess model readiness for deployment
• Strengthen feedback loops from real-world product usage, incorporating both explicit and implicit user signals into post-training

What you need

• Research
• ETL
• Git
• Lean
• M.Sc or PhD or B.Tech or B.E. or B.Sc or BCA

Good to have

• Machine learning