Staff Machine Learning Research Scientist/ Engineer, Agents jobs in United States
cer-icon
Apply on Employer Site
company-logo

Scale AI · 1 day ago

Staff Machine Learning Research Scientist/ Engineer, Agents

Scale AI is a leading AI data foundry focused on accelerating the development of AI applications. They are seeking a Staff Machine Learning Research Scientist/Engineer to explore data types essential for building state-of-the-art agents, guide data strategy, and contribute to impactful research publications while collaborating with customer researchers and engineering teams.

Artificial Intelligence (AI)Data Collection and LabelingGenerative AIImage RecognitionMachine Learning
check
H1B Sponsor Likelynote

Responsibilities

Explore the data landscape needed to advance intelligent, adaptable AI agents
Guide the data strategy at Scale to drive innovation
Contribute to impactful research publications on agents
Collaborate with customer researchers
Work alongside the engineering team to translate advancements into real-world, scalable solutions

Qualification

LLM experiencePytorchResearch publicationJaxTensorflowCloud technologyAgent frameworksAgentic reasoningCommunication skillsCross-functional collaboration

Required

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
Strong written and verbal communication skills and the ability to operate cross-functionally

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

Benefits

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

Company

Scale AI

twittertwittertwitter
company-logo
Scale’s mission is to develop reliable AI systems for the world’s most important decisions.

H1B Sponsorship

Scale AI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (82)
2024 (54)
2023 (29)
2022 (17)
2021 (10)
2020 (10)

Funding

Current Stage
Late Stage
Total Funding
$15.9B
Key Investors
MetaAccelTiger Global Management
2025-06-10Corporate Round· $14.3B
2025-06-04Series Unknown
2024-05-21Series F· $1B

Leadership Team

leader-logo
Jason Droege
Interim Chief Executive Officer
linkedin
leader-logo
Dennis Cinelli
Chief Financial Officer
linkedin
Company data provided by crunchbase