Skip to content
Scale AIScale AISan Francisco, CA

Machine Learning Research Engineer, Agent Data Foundation - Enterprise GenAI

Develops synthetic data pipelines, production trace agents, and automated agent-building frameworks for enterprise GenAI. Requires 3+ years LLM production experience, top conference publications, and advanced CS degree.

218k – 273k/yr
Hybrid3+ YOEML Engineering

About the role

Responsibilities

  • Build synthetic data pipelines to generate enterprise environments for RL post-training.
  • Create agents to convert production traces into actionable insights for agent improvement.
  • Contribute to agent-building product using coding agents and proprietary algorithms.
  • Train state-of-the-art models (internal and community) for enterprise deployment.

Requirements

  • 3+ years building with LLMs in production environments.
  • Experience constructing high-quality data to improve LLMs/Agents.
  • Publications in top conferences (NeurIPS, ICLR, ICML) within last two years.
  • PhD or Master's in Computer Science or related field.

Skills

LLMsReinforcement LearningSynthetic DataAgent BuildingPost-Training AlgorithmsPythonMachine LearningPyTorchData PipelinesTrace Analysis

Similar roles

ML Engineering jobs
Scale AI

ML Systems Engineer, Robotics

Scale AISan Francisco, CA

Designs and builds scalable platforms for serving foundation models in robotics and autonomous vehicles. Requires 4+ years in ML infrastructure, backend systems, GPU optimization, and cloud technologies.

218k – 273k/yr
On-site4+ YOEML Engineering
Scale AI

ML Research Engineer, ML Systems

Scale AISan Francisco, CA +2

Builds and optimizes distributed frameworks for LLM training and inference on Scale's RLXF platform. Collaborates with ML teams to accelerate research, requiring expertise in PyTorch, CUDA, transformers, and large-scale distributed systems.

218k – 273k/yr
On-siteML Engineering
Perplexity

AI Engineer, Applied ML

PerplexitySan Francisco, CA +2

Designs, builds, and iterates on AI/ML models for personalization, query understanding, and content discovery. Requires 5+ years in ML, deep learning expertise (PyTorch/TensorFlow/JAX), Python, and full ML lifecycle ownership.

220k – 405k/yr
On-site5+ YOEML Engineering
Perplexity

AI Inference Engineer

PerplexitySan Francisco, CA +2

Develops APIs and optimizes large-scale ML model inference using Python, Rust, C++, PyTorch, and CUDA. Benchmarks performance, improves reliability, and implements LLM optimizations on GPU architectures.

220k – 405k/yr
On-siteML Engineering
Scale AI

Software Engineer, Enterprise AI

Scale AINew York, NY +1

Build and scale enterprise Generative AI platform, owning large product areas across backend, frontend, LLMs, and ML models. Requires 4+ years experience, proficiency in Python/JavaScript/SQL, Kubernetes, and cloud providers.

216k – 270k/yr
On-site4+ YOEML Engineering