Principal ML Architect: LLMs, Personalization & Scalable AI

hims-and-hers; United States; Remote; USD 180,000 - 260,000; Full time

Benefits offered by this job

Unlimited PTO

Comprehensive health benefits

Equity compensation

Employee Stock Purchase Program (ESPP)

401k with matching

Job summary

Hims & Hers seeks a Principal Machine Learning Engineer to define the architecture for large-scale ML models powering personalized care and treatment intelligence. You will shape platforms and pipelines, enabling secure, scalable training and real-time inference across products like MedMatch.

You will mentor senior engineers, drive cross-functional initiatives with product and clinical teams, and champion strong observability and robust deployment practices in a fast-paced, customer-centric

Qualifications

• 10+ years of ML/engineering with leadership and system architecture experience.
• Expert-level Python and ML frameworks (PyTorch, TensorFlow).
• Proven track record designing and scaling ML platforms for production-grade training and inference.
• Deep experience with LLMs and NLP, including transformer models.
• Strong knowledge of ML infrastructure, monitoring, and observability.
• Experience with ML infrastructure tools such as Databricks, MLFlow, SageMaker, or equivalent.

Responsibilities

• Define architectural vision for ML and LLM systems at scale.
• Establish patterns and practices for scalable, secure, and maintainable systems.
• Prototype and guide implementation of model pipelines using Python.
• Partner with platform, product, and clinical teams to align decisions.
• Evolve ML infrastructure for continuous training and real-time inference.
• Lead cross-functional initiatives to improve velocity and reduce complexity.
• Set quality standards for testing, observability, performance and risk.
• Mentor senior engineers and foster systems-thinking leadership.
• Champion culture of sustainable speed and architectural clarity.

Skills

Python

PyTorch

TensorFlow

LLMs

NLP

ML infrastructure

Education

Master's degree or PhD in Computer Science or related field

Tools

Databricks

MLFlow

SageMaker

Job description

Hims & Hers seeks a Principal Machine Learning Engineer to define the architecture for large-scale ML models powering personalized care and treatment intelligence. You will shape platforms and pipelines, enabling secure, scalable training and real-time inference across products like MedMatch.

You will mentor senior engineers, drive cross-functional initiatives with product and clinical teams, and champion strong observability and robust deployment practices in a fast-paced, customer-centric