Senior / Staff Machine Learning Research Scientist, Agents

Highlights

• Study data types essential for state-of-the-art browser and SWE agents.
• Research agent environments and RL reward signals for autonomous agents.
• Benchmark autonomous agents across real-world scenarios and environments.

What they offer

• base salary
• equity
• benefits

Ideal candidate

Creative ML researcher with 3+ years solving complex problems who thrives turning papers into prototypes and collaborating cross-functionally to advance adaptable, real-world AI agents.

Requirements

• Practical experience working with LLMs, with proficiency in frameworks like PyTorch, Jax, or TensorFlow
• Adept at interpreting research literature and quickly turning new ideas into prototypes
• Track record of published research in top ML venues (e.g., ACL, EMNLP, NAACL, NeurIPS, ICML, ICLR, COLM)
• At least three years of experience addressing sophisticated ML problems in research or product development
• Strong written and verbal communication skills and ability to operate cross-functionally
• Hands-on experience with open source or bespoke LLM fine-tuning using PyTorch/Jax
• Hands-on experience and publications building applications and evaluations for AI agents such as tool-use, text2SQL, browser, coding and GUI agents
• Hands-on experience with agent frameworks such as OpenHands, Swarm, LangGraph
• Familiarity with agentic reasoning methods such as STaR and PLANSEARCH
• Experience with cloud stack (e.g. AWS or GCP) and developing ML models in a cloud environment