Staff Machine Learning Research Scientist/ Engineer, Agents

About Scale AI

At Scale AI, our mission is to accelerate the development of AI applications. This role is at the intersection of cutting-edge AI research and practical application, with a focus on studying the data types essential for building state-of-the-art agents.

Responsibilities

• Explore the data landscape needed to advance intelligent, adaptable AI agents, guiding the data strategy at Scale to drive innovation.
• Contribute to impactful research publications on agents, collaborate with customer researchers, and work alongside the engineering team to translate these advancements into real-world, scalable solutions.

Requirements

• Practical experience working with LLMs, with proficiency in frameworks like Pytorch, Jax, or Tensorflow.
• A track record of published research in top ML venues (e.g., ACL, EMNLP, NAACL, NeurIPS, ICML, ICLR, COLM, etc.).
• At least three years of experience addressing sophisticated ML problems, either in a research setting or product development.

Nice to Have

• Hands-on experience with open source LLM fine-tuning or involvement in bespoke LLM fine-tuning projects using Pytorch/Jax.
• Hands-on experience and publications in building applications and evaluations related to AI agents such as tool-use, text2SQL, browser agents, coding agents and GUI agents.
• Hands-on experience with agent frameworks such as OpenHands, Swarm, LangGraph, etc.
• Familiarity with agentic reasoning methods such as STaR and PLANSEARCH
• Experience working with cloud technology stack (eg. AWS or GCP) and developing machine learning models in a cloud environment.

Benefits

• Comprehensive health, dental and vision coverage
• Retirement benefits
• A learning and development stipend
• Generous PTO
• Commuter stipend

Salary Range

$259,200-$324,000 USD