Staff Machine Learning Engineer (Foundation AI) jobs in United States
cer-icon
Apply on Employer Site
company-logo

WHOOP · 1 day ago

Staff Machine Learning Engineer (Foundation AI)

WHOOP is an advanced health and fitness wearable on a mission to unlock human performance and extend healthspan. The Staff Machine Learning Engineer will drive the research, development, and deployment of large-scale multimodal models to enhance WHOOP’s intelligent and personalized health experiences.

Consumer ElectronicsFitnessSportsWearablesWellness
check
H1B Sponsor Likelynote

Responsibilities

Design, train, and optimize large-scale multimodal foundation models that integrate wearable sensor data, text, biomarkers, and behavioral data
Conduct applied research in self-supervised learning, representation learning, and downstream task fine tuning to advance WHOOP’s core model capabilities
Develop scalable, distributed training pipelines for large models on high-performance compute environments
Collaborate with MLOps, data engineering, and software engineering teams to operationalize models for production deployment, ensuring robustness, reproducibility, and observability
Partner with product and research teams to translate foundation model capabilities into downstream features that deliver meaningful member value
Contribute to the technical roadmap and architectural direction for foundation model development at WHOOP
Serve as a technical mentor for other data scientists, sharing best practices in deep learning, large-scale training, and multimodal data integration
Ensure models adhere to WHOOP’s standards for ethical, transparent, and privacy-preserving AI

Qualification

Deep learning expertiseMultimodal model trainingPython proficiencyDistributed computingMLOps best practicesRepresentation learningSelf-supervised learningPassion for health technologyCommunicationCollaboration skills

Required

Advanced degree (Master's or Ph.D.) in Computer Science, Machine Learning, Electrical Engineering, or a related field, or equivalent professional experience
7+ years of experience in applied ML, AI research, or large-scale modeling, with a track record of delivering production systems
Expertise in modern deep learning (e.g., transformers, state space models) and multimodal model training
Proficiency in Python and deep learning frameworks (e.g., PyTorch, TensorFlow)
Experience building and scaling large datasets and training large models in distributed compute environments
Strong applied experience with representation learning, self-supervised methods, and fine-tuning for downstream applications
Familiarity with MLOps best practices including model versioning, evaluation, CI/CD for ML, and cloud-based compute
Excellent communication skills and ability to collaborate cross-functionally with engineers, researchers, and product teams
Passion for WHOOP's mission to improve human performance and extend healthspan through science and technology

Benefits

Competitive base salaries
Meaningful equity
Benefits
Generous equity package

Company

WHOOP is a fitness tech company that offers wearables that track sleep, recovery, and performance.

H1B Sponsorship

WHOOP has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (32)
2024 (18)
2023 (15)
2022 (5)
2021 (17)
2020 (6)

Funding

Current Stage
Late Stage
Total Funding
$404.75M
Key Investors
SoftBank Vision FundIVPFoundry Group
2021-08-30Series F· $200M
2020-10-28Series E· $100M
2019-11-12Series D· $55M

Leadership Team

leader-logo
Will Ahmed
Founder & CEO
linkedin
leader-logo
Aurelian Nicolae
Co-Founder & Chief Hardware Engineer
linkedin
Company data provided by crunchbase