Skip to content
OpenAIOpenAISan Francisco, CA

Training: ML Framework Engineer

Develops and optimizes internal distributed ML training framework to boost hardware efficiency and enable researchers to experiment with new AI models. Requires strong Python skills, systems understanding, and passion for performance tuning.

205k – 445k/yr
HybridML Engineering

About the role

Responsibilities

  • Apply the latest techniques in our internal training framework to achieve impressive hardware efficiency for our training runs
  • Profile and optimize our training framework
  • Work with researchers to enable them to develop the next generation of models

Requirements

  • Have run small scale ML experiments
  • Love figuring out how systems work and continuously come up with ideas for how to make them faster while minimizing complexity and maintenance burden
  • Have strong software engineering skills and are proficient in Python

Skills

PythonDistributed SystemsMachine LearningPyTorchTensorFlowGpu ProgrammingPerformance OptimizationProfilingSupercomputing

Similar roles

ML Engineering jobs
Perplexity

Forward Deployed Engineer - Applied AI

PerplexityNew York, NY +2

Forward Deployed Engineer designs and deploys AI integrations and agentic workflows into customer enterprise systems, owning end-to-end production rollout. Requires 5+ years software engineering, Python proficiency, and LLM production experience.

205k – 335k/yr
On-site5+ YOEML 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
Kepler

Machine Learning Engineer

KeplerNew York, NY

Build and own ML models, fine-tuning, evaluation harnesses, and routing for Kepler's AI agent harness in finance. Requires 5+ years production software experience and shipped ML systems focused on correctness, evals, and real-world reliability.

200k – 280k/yr
On-site5+ YOEML Engineering
Firecrawl

Research Engineer – Evals

FirecrawlSan Francisco, CA

Build and own evaluation systems, metrics, pipelines, and datasets to rigorously measure the quality of Firecrawl's LLM-ready web data outputs at massive scale, driving model and product improvements.

210k – 275k/yr
Remote4+ YOEML Engineering
AfterQuery

Machine Learning Engineer

AfterQuerySan Francisco, CA

Build production ML systems for measuring, predicting, and scaling data quality for frontier AI models. Requires 3-6 years experience in applied ML or related production systems (ranking, recommendations, data quality, fraud) plus strong software engineering skills.

200k – 300k/yr
On-site3+ YOEML Engineering