# Staff Machine Learning Research Scientist/ Engineer, Agents

**Company**: Scale AI
**Location**: San Francisco, CA; Seattle, WA; New York, NY
**Work arrangement**: hybrid
**Experience**: staff
**Job type**: full-time
**Salary**: $259,200-$324,000 USD
**Category**: Engineering
**Industry**: Technology
**Wikidata**: https://www.wikidata.org/wiki/Q112629176

**Apply**: https://job-boards.greenhouse.io/scaleai/jobs/4488520005?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_5aa5b947-f4d

## Description

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

## Skills

### Required
- Pytorch
- Jax
- Tensorflow
- LLMs
- Agent frameworks
- Agentic reasoning methods
- Cloud technology stack

### Nice to have
- Open source LLM fine-tuning
- Bespoke LLM fine-tuning projects

---

Source: [Apply at job-boards.greenhouse.io](https://job-boards.greenhouse.io/scaleai/jobs/4488520005?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)