Latest AI JobsBrowse Jobs
Scale AI logo

Senior / Staff Machine Learning Research Scientist, Agents

Scale AISan Francisco, CA; Seattle, WA; New York, NYResearch
Apply on company site
Salary
Not listed
Job type
Full-time
Posted
25 Oct 2024
Status
Open · checked 7d ago

About this role

The Senior or Staff Machine Learning Research Scientist, Agents position at Scale AI is based in San Francisco, CA, Seattle, WA, or New York, New York, and suits professionals experienced in advanced machine learning and LLM agent research. This role sits within Scale AI's Research organization on the Agent Capabilities & Environments team, focusing on the data landscape required to advance intelligent and adaptable AI agents.

Role Overview

This position exists at the intersection of practical application and advanced AI research, requiring a focus on data types essential for building state-of-the-art browser and SWE agents. The ideal professional will guide the data strategy at Scale AI to drive innovation, explore data landscapes needed for intelligent agents, and contribute to impactful research publications. Team members work alongside engineers and customer researchers to translate advancements into scalable, real-world solutions.

Key Responsibilities

Study data types essential for building state-of-the-art agents such as browser and software engineering agents.
Guide the organization's data strategy to drive innovation in intelligent, adaptable AI agents.
Contribute to impactful research publications regarding autonomous agents.
Collaborate closely with customer-facing researchers and applied AI engineers.
Translate research advancements into real-world, scalable deployment solutions.

Requirements

Practical experience working with large language models and proficiency in frameworks like PyTorch, Jax, or TensorFlow.
Adeptness at interpreting research literature and swiftly translating fresh ideas into working prototypes.
A documented track record of published research in top machine learning venues such as NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, or COLM.
At least three years of experience solving sophisticated machine learning problems within a product development or research environment.
Strong written and verbal communication skills with the capability to operate cross-functionally.

Nice to Have

Hands-on experience with open-source LLM fine-tuning or involvement in bespoke tuning projects using PyTorch or Jax.
Publications or hands-on experience in building applications and evaluations for AI agents like tool-use, browser, coding, GUI, and text2SQL agents.
Familiarity with agentic reasoning methods such as PLANSEARCH and STaR.
Experience utilizing agent frameworks including LangGraph, Swarm, or OpenHands.
Background working with cloud technology stacks like AWS or GCP to develop machine learning models in a cloud ecosystem.

Skills and Tools

PyTorch
TensorFlow
Jax
AWS
GCP
LLMs
Agent frameworks

Salary and Benefits

The base salary range for this full-time position located in San Francisco, New York, or Seattle is $302,400 to $378,000 USD. Eligible roles include equity-based compensation subject to Board of Director approval, alongside comprehensive health, dental, and vision coverage, retirement benefits, learning and development stipends, a commuter stipend, and generous paid time off.

About the Company

Scale AI acts as a leading AI data foundry, fueling advancements in generative AI, defense applications, and autonomous vehicles with high-quality data and full-stack technologies. The company collaborates with industry leaders and government agencies, including Meta, the Mayo Clinic, Ernst & Young, and the U.S. Air Force, to help build and oversee reliable AI systems.

How to Apply

To submit an application for this opportunity, please visit the official Scale AI careers page.

Skills and tech stack

Questions candidates ask

What locations are eligible for this role?

This position is located in San Francisco, CA, Seattle, WA, and New York, NY.

What is the base salary range for this position?

The base salary range for this full-time role in San Francisco, New York, and Seattle is $302,400 to $378,000 USD.

What qualifications are needed regarding research publications?

Candidates should have a track record of published research in top machine learning venues such as NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, or COLM.

What benefits are included with employment?

Benefits include comprehensive health, dental, and vision coverage, retirement benefits, equity compensation, learning and development stipends, generous PTO, and a potential commuter stipend.

What team does this role belong to?

This role belongs to the Agent Capabilities & Environments team within Scale's Research organization.

Ready to apply?

The application is on the official Scale AI careers page.

Apply on company site