Skip to content
EightsleepEightsleepSan Francisco, CA

Machine Learning Engineer (Foundation Models & Personalization)

Builds and deploys ML models for sleep personalization, readiness forecasting, and behavior prediction using foundation models and multimodal data. Requires 2+ years production ML experience with PyTorch/TensorFlow, personalization systems, and strong Python engineering.

Salary not listed
On-site2+ YOEML Engineering

About the role

Responsibilities

  • Build and deploy ML models that improve sleep experiences through personalization, prediction, and behavior understanding (e.g., readiness forecasting, event detection, individualized recommendations).
  • Apply and adapt foundation-model capabilities to real product workflows (LLM + tools/RAG, multimodal modeling, policy learning), including MCP-style integrations where helpful.
  • Develop user behavior models that connect longitudinal signals (sleep, environment, routines) to actionable interventions - grounded in robust experimentation and measurement.
  • Design evaluation strategies (offline metrics, slice-based analysis, calibration, reliability, fairness) and partner with Product to run high-quality online experiments.
  • Productionize models: scalable training/inference pipelines, model monitoring, drift detection, alerting, and continuous improvement loops.
  • Collaborate with cross-functional partners (Product, Mobile, Backend, Clinical) to scope requirements and ship high-impact features.

Requirements

Minimum Qualifications

  • 2+ years building ML systems in production, ideally for consumer-facing products.
  • Strong ML fundamentals across supervised learning, sequence/time-series modeling, and modern deep learning.
  • Hands-on experience with large-scale model training and evaluation (PyTorch/TensorFlow/JAX), and strong Python engineering practices.
  • Experience with personalization systems (ranking/recommendations, segmentation, lifecycle modeling, propensity/behavior modeling, causal/experiment-aware thinking).
  • Fluency with data tooling (SQL, distributed compute such as Spark/Ray, and cloud storage/compute).
  • Strong product sense: you can translate ambiguous goals into measurable outcomes and iterate quickly with stakeholders.

Bonus Points

  • Experience applying LLMs/foundation models to product features (tool use, retrieval, structured outputs, guardrails, evals).
  • Experience with multimodal data (sensor signals + context) and/or health/biometrics data.
  • Experience with privacy-preserving approaches (on-device/federated learning, differential privacy, data minimization).
  • Experience designing experimentation frameworks or causal inference approaches for personalization.

Skills

PyTorchTensorFlowJAXPythonSQLSparkRayLLMsRAGFederated Learning

Similar roles

ML Engineering jobs
Nuro

Software Engineer, Performance

NuroMountain View, CA

New grad software engineer optimizing performance of autonomous vehicle software. Profile, debug, and reduce latency of C++ systems running on x86, ARM, and GPUs while building tools for memory management and high-performance code.

153k – 175k/yr
On-siteEntry levelML Engineering
Airbnb

Machine Learning Engineer, Relevance and Personalization

AirbnbUnited States

Build and productionize cutting-edge ML models and ranking algorithms for Airbnb's search, recommendation, and personalization systems at scale. Requires PhD (new grad) or 2+ years applied ML experience with strong programming and ML systems skills.

166k – 195k/yr
Remote2+ YOEML Engineering
Abridge

Software Engineer

AbridgeSan Francisco, CA +1

Early-career Software Engineer building and iterating on agentic LLM systems, RAG pipelines, evaluation frameworks, and backend/frontend components for AI-powered clinical documentation at Abridge. Requires CS degree or equivalent plus hands-on GenAI experience from projects, internships or coursework.

157k – 184k/yr
On-siteEntry levelML Engineering
Mirage

ML Engineer, Generative Video

MirageNew York, NY

Build and scale video generation models at Mirage. Optimize training/inference for low-latency, real-time performance using PyTorch, CUDA, and distributed systems. Requires 2+ years industry experience in deep learning infrastructure.

175k – 275k/yr
On-site2+ YOEML Engineering
Traba

Software Engineer

TrabaNew York, NY

Build and own production AI agent systems (harnesses, evals, orchestration) on frontier LLMs for industrial supply chain workflows at Traba. Requires 1+ years shipping LLM/agent features to production, strong Python/TS skills, and high-agency in ambiguous customer environments.

140k – 200k/yr
On-site1+ YOEML Engineering