Staff AI Engineer — On-Device Models & Training

Ghost

San Francisco (CA)

On-site

USD 180,000 - 240,000

Full time

5 days ago
Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Ghost is seeking a specialist to advance post-training language models that run locally on Core hardware. You will post-train models, develop reinforcement learning methods, and shape on-device agent behavior with an emphasis on latency and privacy.

You should have strong Python and PyTorch skills, with experience building training pipelines, datasets, and evaluations, and be comfortable working on-site in San Francisco with cross-functional teams.

Qualifications

  • Experience post-training language models with supervised fine-tuning, reinforcement learning, or preference optimization.
  • Strong Python and PyTorch skills, with experience building and debugging training pipelines.
  • Experience creating datasets and evaluations that produce measurable improvements in model behavior.
  • Familiarity with tool-using agents, multi-step environments, and the challenges of training against task outcomes.
  • Strong experimental judgment: you can diagnose failures, test hypotheses, and distinguish benchmark gains from meaningful product improvements.
  • Ability to work on-site in San Francisco.

Responsibilities

  • Post-train open models to power personal agents that run locally on Core.
  • Improve tool use, reasoning, and reliable execution across complex, multi-step tasks.
  • Build training datasets and feedback pipelines grounded in realistic agent workflows.
  • Develop supervised fine-tuning and reinforcement learning methods that translate model capability into useful product behavior.
  • Build evaluations that measure task completion, personalization, and when agents should act or ask for guidance.
  • Work closely with the agent and inference teams to improve quality within on-device memory, compute, and latency constraints.

Skills

Python
PyTorch
On-device inference

Tools

PyTorch

Job description

Ghost is seeking a specialist to advance post-training language models that run locally on Core hardware. You will post-train models, develop reinforcement learning methods, and shape on-device agent behavior with an emphasis on latency and privacy.

You should have strong Python and PyTorch skills, with experience building training pipelines, datasets, and evaluations, and be comfortable working on-site in San Francisco with cross-functional teams.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Member of Technical Staff - Post-Training
Member of Technical Staff - Post-Training

Ghost • San Francisco (CA)

On-site
USD 180,000 - 240,000
Staff Inference Engineer — Local AI & Performance
Staff Inference Engineer — Local AI & Performance

Ghost • San Francisco (CA)

On-site
USD 180,000 - 240,000
Member of Technical Staff - Inference
Member of Technical Staff - Inference

Ghost • San Francisco (CA)

On-site
USD 180,000 - 240,000
Staff Engineer, Physical AI & Video Modeling
Staff Engineer, Physical AI & Video Modeling

Build AI • San Francisco (CA)

On-site
USD 230,000 - 260,000
Competitive pay
Medical, dental, and vision packages
Housing subsidy
+6
Staff ML Engineer - Post-Training & Model Foundry
Staff ML Engineer - Post-Training & Model Foundry

San Francisco Tensor Company • San Francisco (CA)

On-site
USD 275,000 - 315,000
Relocation assistance
On-Device AGI Researcher — Train & Ship Trusted Models
On-Device AGI Researcher — Train & Ship Trusted Models

AGI, Inc. • San Francisco (CA)

On-site
USD 180,000 - 280,000
AI Engineer: Model Training, Inference & GPU Infra
AI Engineer: Model Training, Inference & GPU Infra

Agentrys • San Jose (CA)

Hybrid
USD 170,000 - 210,000
Research Engineer/Research Scientist- Personal AGI, Model Experience
Research Engineer/Research Scientist- Personal AGI, Model Experience

OpenAI • San Francisco (CA)

On-site
USD 380,000 - 500,000
Relocation assistance
Agent Post-Training & Artifacts Researcher
Agent Post-Training & Artifacts Researcher

OpenAI • San Francisco (CA)

On-site
USD 120,000 - 160,000
Health benefits
Flexible work hours
Professional development opportunities
Principal Machine Learning Engineer
Principal Machine Learning Engineer

LE130 Unity Technologies SF • Mountain View (CA)

On-site
USD 260,000 - 380,000