Skip to content
Scale AIScale AISan Francisco, CA

Staff Machine Learning Research Scientist/ Engineer, Agents

Conducts cutting-edge research on data for state-of-the-art AI agents like browser and SWE agents, develops prototypes using LLMs and frameworks like PyTorch/JAX, and publishes in top ML venues. Requires 3+ years ML experience and strong cross-functional communication.

275k – 350k
On-siteAI Research

About the role

About This Role

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 expertise in LLM agents and planning algorithms, 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.

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.

Skills

PyTorchJAXTensorFlowLLMsAI AgentsLangGraphOpenhandsSwarmAWSGCP

Similar roles

AI Research jobs
Datadog

Staff Applied Scientist - Agentic Interfaces

DatadogNew York, NY

As a Staff Applied Scientist, you will define and build measurement systems for AI agent interfaces at Datadog, focusing on evaluation strategy, metric definition, and dataset creation to improve agent performance on customer workflows.

276k – 345k
On-site10+ YOEAI Research
Databricks

Sr. Staff AI Research TLM - AI Systems

DatabricksMountain View, CA +1

Lead a world-class research team advancing LLM scaling, post-training, RL, and inference efficiency at Databricks AI. Drive research roadmap and translate breakthroughs into production systems while collaborating closely with engineering and product teams.

270k – 340k
On-siteAI Research
Runway

Member of Technical Staff, Research Engineer

RunwayUnited States

Research Engineer develops and experiments with multimodal world models, focusing on data strategies, training techniques, evaluations, and production deployment for AI simulation technologies. Requires 4+ years in ML research/engineering and proficiency in PyTorch or JAX.

270k – 370k
Remote4+ YOEAI Research
Runway

Member of Technical Staff, Research Engineer (GPU Performance)

RunwayUnited States

Research Engineer optimizes GPU performance for large-scale AI world model training and real-time inference, focusing on custom kernels, distributed systems, and mixed-precision techniques. Requires 4+ years in ML infrastructure and performance optimization.

270k – 370k
Remote4+ YOEAI Research
Runway

Member of Technical Staff, Applied Research Scientist

RunwayUnited States

Applied Research Scientist develops computer vision and generative models for creative AI tools, collaborating cross-functionally to productionize research. Requires 4+ years in ML/CV/graphics, strong Python/C++/PyTorch skills, and experience mentoring researchers.

260k – 370k
Remote4+ YOEAI Research