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.
About the job
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
PyTorch, TensorFlow, JAX, Python, SQL, Spark, Ray, LLMs, RAG, Federated Learning
Similar jobs
ML Engineering jobs
DataVisor
Mountain View, CA
Builds high-scale data pipelines, distributed systems, and AI agent workflows using LLMs for fraud intelligence platform. Requires 2+ years software engineering, Python proficiency, big data tools, AWS/K8s, and ML foundations.
Coinbase
San Francisco, CA
Machine learning intern pursuing a Ph.D. who will build and deploy production-scale models and pipelines, conduct an end-to-end research project, and collaborate with engineering and product teams on blockchain and cryptocurrency applications.
Fireworks AI
San Mateo, CABuild, deploy, and optimize AI applications and machine learning models for customer use cases while contributing to an internal ML platform. This new graduate role requires a technical master’s degree, hands-on ML or LLM experience, and strong customer communication skills.
PathAI
Boston, MADevelop and deploy machine learning models for biomedical research and AI product development, collaborating with scientific, engineering, and product teams. Requires a master's degree with 2–4 years of experience or a PhD with 0–2 years, plus strong Python and ML development skills.
Earnin
Mountain View, CA
Machine learning engineer who trains, evaluates, and productionizes models and LLM-powered applications for financial products. Requires 2+ years of ML systems experience, strong Python and PyTorch skills, production data pipelines, model evaluation, and API development.