Senior AI Engineer

Citizen Health; San Francisco (CA); On-site; USD 150,000 - 300,000; Full time

Benefits offered by this job

Competitive salary + equity package

Comprehensive health, dental, and vision insurance

Unlimited paid time off and generous parental leave

Flexible hybrid work environment

Job summary

A leading healthcare technology company in San Francisco is seeking an AI/Machine Learning Engineer to design and deploy AI solutions that transform health data into valuable insights. You will lead projects that directly impact patient care, working within a supportive culture dedicated to diversity and innovation. Ideal candidates will have extensive experience in machine learning, strong Python skills, and familiarity with LLM frameworks. This role offers a competitive salary and a flexible hybrid work environment.

Qualifications

• 5+ years of experience in machine learning engineering with a focus on production ML systems.
• Strong proficiency in Python and AI/LLM frameworks.
• Experience designing and implementing production-grade ML pipelines.

Responsibilities

• Design and implement end-to-end machine learning solutions.
• Develop and optimize LLM solutions for healthcare applications.
• Create robust data pipelines for medical data.

Skills

Machine learning engineering

Production ML systems

Python

AI/LLM frameworks (LangChain, LangGraph, AutoGen, CrewAI)

Data preprocessing

MLOps

Solid software engineering skills

Education

Degree in Computer Science or related field

Tools

LangChain

LangGraph

AutoGen

Job description

Overview

Citizen Health is building an AI‑driven patient‑centric platform that uses patients' complete medical histories to produce actionable insights for clinical decision‑making and day‑to‑day care.

Role

As an AI/Machine Learning Engineer you will lead the design and deployment of advanced AI solutions that transform complex health data into high‑value insights for patients, clinicians, and researchers.

Responsibilities

• Design and implement end‑to‑end machine learning solutions—from data preprocessing to model deployment and monitoring.
• Develop and optimize Large Language Model (LLM) solutions for healthcare applications, including fine‑tuning and Retrieval‑Augmented Generation (RAG).
• Create robust data pipelines for validation and deployment of medical data (structured clinical data, medical imaging, unstructured text).
• Collaborate with backend engineers to integrate ML models into production infrastructure.
• Ensure ML systems meet strict healthcare compliance requirements while maintaining high performance.
• Implement monitoring systems for model performance, data drift, and system health.
• Lead research initiatives to explore and apply state‑of‑the‑art AI/ML techniques.
• Collaborate with clinical experts to validate model outputs and ensure medical accuracy.

Must‑Have Skills

• 5+ years of experience in machine learning engineering with a focus on production ML systems.
• Strong proficiency in Python and AI/LLM frameworks (LangChain, LangGraph, AutoGen, CrewAI).
• Experience designing and implementing production‑grade ML pipelines.
• Proven track record of deploying ML/LLM solutions in production environments.
• Solid understanding of ML fundamentals, including deep learning, NLP, and statistical modeling.
• Familiar with MLOps tools and best practices.
• Strong software engineering skills and coding best practices.
• Degree in Computer Science or a related field.

Preferred Skills

• Hands‑on experience integrating and optimizing LLMs in production environments.
• Proficiency with LLM tools and techniques such as LangChain, LlamaIndex, Prompt Engineering.
• Experience with Retrieval‑Augmented Generation (RAG) systems.
• Knowledge of vector databases and efficient similarity search techniques for AI applications.
• Familiarity with message queuing systems and data streaming platforms.
• Strong architectural design and coding quality.
• Passion for building robust, scalable AI‑driven systems in healthcare.

Why Join Us

• Work on pioneering AI technologies that push the boundaries of healthcare.
• Lead projects with direct impact on patient outcomes.
• Fast‑paced, high‑growth environment offering clear career progression.
• Supportive culture of learning, experimentation, and innovation.
• Commitment to diversity and inclusion with an inclusive work environment.
• Regular team activities and knowledge sharing.

Perks

• Competitive salary + equity package.
• Comprehensive health, dental, and vision insurance.
• Unlimited paid time off and generous parental leave.
• Flexible hybrid work environment.

Compensation

U.S. / San Francisco, CA – $150 K – $300 K annual.

Diversity & Inclusion

Citizen Health is an equal opportunity employer committed to creating an inclusive environment for all employees. We welcome applicants of all backgrounds, identities, and experiences.

Apply

We welcome candidates with the right passion and mindset, even if you don’t meet every qualification. If you’re excited about the mission, apply now.

Referrals

Referrals increase your chances of interview selection by 2×.