Remote Staff AI Engineer, Research Flow Lead

Remote

San Francisco (CA)

Remote

USD 210,000 - 320,000

Full time

14 days+

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Job summary

Typeform is seeking a Staff AI Engineer to shape the technical direction of Research Flow and the AI systems behind our products. You will own complex engineering challenges, connect AI capabilities into reliable customer experiences, and drive architecture across generative AI, enterprise RAG, and model evaluation at scale.

You will lead production ML pipelines, build reusable services, and guide decisions on build vs.

Qualifications

  • Significant experience building and operating ML/AI systems in production.
  • Track record leading complex engineering initiatives across teams.
  • Strong Python and software engineering skills with production code experience.
  • Experience designing production services and APIs (e.g., FastAPI).
  • Experience with AWS, Docker, Kubernetes and cloud-based infrastructures.
  • Experience with ML lifecycle tools and observability for production systems.

Responsibilities

  • Shape the technical direction of Research Flow and AI systems.
  • Lead architectural decisions across AI-assisted study design and research synthesis.
  • Own ambiguous problems from exploration to production delivery.
  • Build reusable services and APIs for AI capabilities.
  • Define evaluation strategies and release criteria for generative AI applications.
  • Mentor engineers and raise engineering standards across teams.

Skills

Python
Software engineering
Production ML/AI systems
Leadership beyond own projects
API design / FastAPI
AWS
Docker
Kubernetes
LangChain / LangGraph
PyTorch
Observability (Datadog/OpenSearch)
Mentoring engineers
Cross-team collaboration

Tools

Docker
Kubernetes
AWS
Terraform
Airflow
Kafka
MLflow
Datadog
OpenSearch
SageMaker/Bedrock

Job description

Typeform is seeking a Staff AI Engineer to shape the technical direction of Research Flow and the AI systems behind our products. You will own complex engineering challenges, connect AI capabilities into reliable customer experiences, and drive architecture across generative AI, enterprise RAG, and model evaluation at scale.

You will lead production ML pipelines, build reusable services, and guide decisions on build vs.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Staff AI Engineer — AI Systems & Research Flow Lead
Staff AI Engineer — AI Systems & Research Flow Lead

Typeform • United States

Remote
USD 170,000 - 210,000
Staff AI Engineer — Remote, AI Systems Leader
Staff AI Engineer — Remote, AI Systems Leader

Typeform • Northern (KY)

Hybrid
USD 170,000 - 210,000
Staff AI Engineer – Remote, Generative AI & RAG
Staff AI Engineer – Remote, Generative AI & RAG

The available sources do not contain information about the company name for rounx.com. • United States

Remote
USD 170,000 - 210,000
Fully remote
Competitive salary
Performance-based bonus
Senior AI Engineer - Remote EU
Senior AI Engineer - Remote EU

Typeform • United States

Remote
USD 120,000 - 180,000
Remote Senior AI Engineer: Generative AI & RAG Lead
Remote Senior AI Engineer: Generative AI & RAG Lead

Kenyan Edu • United States

Remote
USD 150,000 - 190,000
Remote Senior AI Engineer: Scalable GenAI & RAG Systems
Remote Senior AI Engineer: Scalable GenAI & RAG Systems

Typeform • Northern (KY)

Hybrid
USD 150,000 - 210,000
Senior AI Engineer - US
Senior AI Engineer - US

Typeform • United States

Hybrid
USD 150,000 - 210,000
Staff AI Systems Engineer
Staff AI Systems Engineer

Flow Engineering • San Francisco (CA)

On-site
USD 190,000 - 270,000
Senior AI-Driven Full-Stack Engineer
Senior AI-Driven Full-Stack Engineer

Flow Engineering, Inc. • San Francisco (CA)

On-site
USD 190,000 - 260,000
Health coverage
Dental coverage
Vision coverage
+2
Senior AI Engineer - US
Senior AI Engineer - US

Kenyan Edu • United States

Remote
USD 150,000 - 190,000