Menu
Full-Time

Machine Learning Research Engineer, Agent Data Foundation – Enterprise GenAI at Scale

Company Scale
Location San Francisco, CA; New York, NY
Salary Competitive salary
Posted Posted on January 2, 2026

Job Description

Opening. We are working on an arsenal of proprietary research, tools, and resources that serve all of our enterprise clients. As MLRE on the Data Foundation team, you’ll work on cutting edge research to define the data flywheel that makes the whole machine move.

What you'll do

You will build synthetic data pipelines to generate enterprise environments to use for RL post-training, create agents to convert traces from production into actionable insights to use to improve agents, contribute to our agent building product which can construct other agents using coding agents + proprietary algorithms, and train state of the art models, developed both internally and from the community, to deploy to our enterprise customers.

What you need

  • 3+ years of building with LLMs in a production environment
  • Clear experiences with constructing high quality data to use to improve an LLM/Agent
  • Publications in top conferences such as NEURIPS, ICLR, or ICML within the last two years
  • PhD or Masters in Computer Science or a related field

Similar Jobs

Full-Time

Forward Deployed Engineer

Unitary
More Info
Full-Time

Applied Scientist II, Foundation Model, Industrial Robotics Group

Amazon
Sunnyvale
More Info
Full-Time

Manager, Tech Program Management

Amazon
Los Angeles County
More Info
Full-Time

Sr. Financial Analyst, Finance – WFM Merch Partnerships

Amazon
Austin, TX
More Info
Full-Time

Sr. Financial Analyst, Amazon Security Finance

Amazon
More Info
Apply Now