The Role

We're seeking a Senior Machine Learning Engineer to lead the development and deployment of cutting-edge AI solutions that power our patient-centric platform. This role is crucial in designing and implementing sophisticated machine learning models that transform complex health data into actionable insights for patients, healthcare providers, and researchers.

As a key technical leader, you'll be at the forefront of applying advanced machine learning techniques to solve challenging problems in rare disease research and patient care. Your work will be instrumental in developing AI-driven solutions that improve disease understanding, treatment recommendations, and patient outcomes.

This is a hybrid role based out of our office in San Mateo, CA and requires 2-3 days in office.

Responsibilities

• Design and implement end-to-end machine learning solutions, from data preprocessing to model deployment and monitoring
• Develop and optimize agentic Large Language Models (LLMs) solutions for healthcare applications using techniques like fine-tuning and Retrieval-Augmented Generation (RAG)
• Create robust data pipelines for validation, and deployment
• Implement ML systems that can effectively process and analyze diverse healthcare data types, including structured clinical data, medical imaging, and unstructured text
• Collaborate with backend engineers to integrate ML models into our production infrastructure
• Ensure ML systems meet strict healthcare compliance requirements while maintaining high performance
• Design and implement monitoring systems for model performance, data drift, and system health
• Lead research initiatives to explore and implement state-of-the-art AI/ML techniques
• Collaborate with clinical experts to validate model outputs and ensure medical accuracy
• Stay current with the latest developments in AI/ML, particularly in healthcare applications

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
• Strong understanding of ML fundamentals, including deep learning, NLP, and statistical modeling
• Experience with ML ops tools and best practices
• Proficiency in working with large-scale datasets and distributed computing
• Strong software engineering skills and coding best practices

Preferred Skills

• Hands-on experience integrating and optimizing Large Language Models (LLMs) in production environments
• Proficiency with LLM tools and techniques such as LangChain, LlamaIndex, Prompt Engineering
• Experience implementing Retrieval-Augmented Generation (RAG) systems
• Familiarity with vector databases and efficient similarity search techniques for AI applications
• Experience with message queuing systems and data streaming platforms
• Familiarity with data processing and analytics technologies
• Proven track record of high-quality coding and architectural design
• Passion for creating robust, scalable, AI-driven systems that make a real-world impact in healthcare

At Citizen Health, we are committed to fostering a diverse and inclusive workplace. We are an equal opportunity employer and welcome candidates from all backgrounds to apply. We particularly encourage applications from individuals with personal or family experiences with rare diseases, as your insights could be invaluable to our mission.

Join us in revolutionizing rare disease research and patient care through the power of technology!

California Base Pay Range: $170,000 - $210,000

This salary range is an estimate, and the actual salary may vary based on a wide range of factors, including your skills, qualifications, experience and location. This position is eligible for stock options and other benefits including but not limited to Medical, Dental, Vision insurance. We also offer unlimited PTO and a paid new parent leave program so you can spend time with your new child.