Skip to content
Scale AIScale AISan Francisco, CA

Staff Machine Learning Research Engineer, Agent Post-training - Enterprise GenAI

Develops next-gen Agent RL training platform for enterprise GenAI, integrating cutting-edge research to train state-of-the-art models for complex use cases. Requires 5+ years LLM production experience, RLHF expertise, recent top publications, and advanced CS degree.

218k – 273k
On-site5+ YOEAI Research

About the role

Responsibilities

  • Train state-of-the-art models (internal and community-developed) for enterprise customers.
  • Research and integrate cutting-edge algorithms into the training stack.
  • Design solutions for complex multi-agent systems to learn from process and outcome-based rewards.

Requirements

  • 5+ years of LLM training in production environments.
  • Experience with post-training methods (RLHF/RLVR) and algorithms (PPO/GRPO).
  • Publications in top conferences (NeurIPS, ICLR, ICML) within last 2 years.
  • PhD or Master's in Computer Science or related field.

Skills

Llm TrainingRLHFRlvrPpoGrpoMulti-Agent SystemsReinforcement LearningPyTorchKubernetesMachine Learning

Similar roles

AI Research jobs
Duolingo

Staff AI Research Engineer

DuolingoNew York, NY +1

Build and deploy AI systems (LLMs, bandits) for monetization, balancing revenue and retention. Requires advanced ML degree or equivalent, applied large-model experience, and leadership skills.

221k – 331k
On-site7+ YOEAI Research
Writer

Staff AI Research Scientist

WriterSan Francisco, CA +2

Lead high-impact research on LLMs and agentic systems, driving post-training, reasoning, and evaluation to power enterprise AI deployments. Requires 7+ years ML research experience, PhD or equivalent, and strong publication record.

234k – 296k
Hybrid7+ YOEAI Research
Snowflake

Staff Applied Research Scientist

SnowflakeMenlo Park, CA

Lead applied research applying formal methods, automated reasoning, and AI (incl. LLMs) to improve correctness, reliability and velocity of Snowflake's cloud data platform and distributed systems. Translate ideas into production capabilities; requires PhD + 8+ years experience and track record of impact.

236k – 339k
On-site8+ YOEAI Research
Snowflake

Staff Research Scientist, Exotic AI

SnowflakeBellevue, WA

Build next-generation training infrastructure for physical AI models that perceive, reason, and act in structured environments. Lead development of representation models, latent world models, and policy optimization systems.

236k – 339k
On-site8+ YOEAI Research
Snowflake

Staff Research Scientist, AI Agents & LLMs

SnowflakeBellevue, WA +1

Leads research in agentic AI and LLMs, developing models for enterprise reasoning, autonomous agents with tool use, and production systems. Requires PhD, expertise in LLM training/fine-tuning, agent systems, and technical leadership.

236k – 339k
HybridAI Research