Skip to content
PolymathPolymathSan Francisco, CA

Software Engineer

Build simulation environments, tasks, and verifiers to train and evaluate long-horizon autonomous AI agents using reinforcement learning. Requires strong engineering fundamentals, high agency, and focus on robust systems.

Salary not listed
RemoteML Engineering

About the role

Responsibilities

  • Build diverse, high-fidelity environments that test agents in realistic settings
  • Design complex tasks that require long-horizon reasoning and tool use
  • Develop robust verifiers that reliably measure agent performance
  • Improve infrastructure and tooling to run, debug, and improve environments
  • Work closely with the research team to identify failure modes and turn them into new tasks and benchmarks

Requirements

  • Strong engineering fundamentals
  • Enjoy building from first principles and solving open-ended technical problems
  • High agency and a strong bias toward shipping
  • High quality bar and care about building robust systems

Skills

Reinforcement LearningSimulation EnvironmentsPythonInfrastructureToolingDebuggingVerifiersBenchmarks

Similar roles

ML Engineering jobs
Cerebras Systems

CoDesign & NextGen Performance Engineer

Cerebras SystemsSunnyvale, CA

Characterize, analyze, and optimize performance of state-of-the-art AI models on Cerebras' wafer-scale hardware. Build performance models, optimize kernels and compilers, debug runtime behavior, and develop visualization tools to influence next-gen AI architecture.

Salary not listed
On-site3+ YOEML Engineering
OpenAI

Research Engineer, Privacy

OpenAISan Francisco, CA

Research Engineer on OpenAI's Privacy team designing and prototyping privacy-preserving ML algorithms like differential privacy and federated learning at scale. Requires hands-on PETs experience, fluency in PyTorch/JAX, and a track record implementing or publishing novel privacy work.

380k – 445k/yr
HybridML Engineering
Console

Research Engineer

ConsoleSan Francisco, CA

Research Engineer building self-improving AI agent systems at Console. Develop eval/optimization loops, fine-tune specialist models, and improve agent reasoning over enterprise context using production data to drive measurable gains in quality, latency, and reliability.

200k – 350k/yr
On-siteML Engineering
Notion

Software Engineer, AI Platform

NotionSan Francisco, CA +1

Build and scale the shared AI platform foundations at Notion, enabling fast and safe shipping of AI products. Requires experience with LLM/ML platforms, strong ownership, and comfort across backend, infrastructure, and product code.

180k – 201k/yr
Hybrid5+ YOEML Engineering
Liftoff

Machine Learning Engineer

LiftoffCalifornia

Machine Learning Engineer building statistical models, optimization systems, and experiments for mobile ad tech economics on the Revenue Engine team. Requires PhD in CS/ML/Economics and industry experience applying ML or economics at scale.

215k – 275k/yr
RemoteML Engineering