Senior / Staff Machine Learning Research Scientist, AgentsActive$46K–$183K

The opportunity

At Scale AI, our mission is to accelerate the development of AI applications. For 8 years, Scale has been the leading AI data foundry, helping fuel the most exciting advancements in AI, including: generative AI, defense applications, and autonomous vehicles.

What you'll do

  • 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.

  • 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.

What they're looking for

  • 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.