Senior / Staff Machine Learning Research Scientist, Agents

San Francisco, CAfull timeOn Site

Apply on Employer Site

senior

$218k–$273k

H1B: yes

Scale AI Senior / Staff Machine Learning Research Scientist role focused on advancing intelligent AI agents through research and data strategy. Responsibilities include studying data types for agents, collaborating on research publications, and translating advancements into scalable solutions.

Responsibilities

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, such as browser and SWE agents. The ideal candidate will explore the data landscape needed to advance intelligent, adaptable AI agents, guiding the data strategy at Scale to drive innovation. This position requires not only expertise in LLM agents and planning algorithms but also creativity in addressing novel challenges related to data, interaction, and evaluation. You will 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.
Ideally you’d have:
Practical experience working with LLMs, with proficiency in frameworks like Pytorch, Jax, or Tensorflow. You should also be adept at interpreting research literature and quickly turning new ideas into prototypes.
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.
Strong written and verbal communication skills and the ability to operate cross-functionally.

Qualification

Practical experience working with LLMsProficiency in frameworks like Pytorch

Required

Practical experience working with LLMs
Proficiency in frameworks like Pytorch, Jax, or Tensorflow
Track record of published research in top ML venues
At least three years of experience addressing sophisticated ML problems
Strong written and verbal communication skills

Preferred

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.
Our research interviews are crafted to assess candidates' skills in practical ML prototyping and debugging, their grasp of research concepts, and their alignment with our organizational culture. We will not ask any LeetCode-style questions.

Benefits

Dental
Vision
PTO
Equity/RSU

Company

Scale AI

View company page

Source: greenhouse

Active within the past hour

More roles at Scale AI

Tech Lead Manager- MLRE, ML Systems

San Francisco, CA

SWE Fellow - Human Frontier Collective (Canada)

Remote (Growth only)

Subject Matter Expert

Remote (Growth only)

Strategic Projects Lead, Generative AI

San Francisco, CA

Boost Your Interview Chances

Improve Resume Match Score

Free

Your Score

9.0

Strong Applicant

Must-Have Skills for This Role

LLMsPytorchJaxTensorflowplanning algorithmsOpenHan

Optimize my Resume

Apply Faster with Autofill

Free

Save your profile once. LumiJobs pre-fills the employer's application form so you stop re-typing the same answers on every site.

Apply With Autofill

Or skip LumiJobs's tools and submit directly on the employer's career site.

Apply on Employer Site

Senior / Staff Machine Learning Research Scientist, Agents at Scale AI · LumiJobs