AI/Machine Learning Engineer

Who We AreCitizen Health was founded on the belief — shaped by firsthand lived experiences navigating healthcare — that having the right advocate is the single most important factor in achieving better care and outcomes. By uniquely combining data, AI, and community, Citizen is building a personalized AI advocate powered by patients' complete medical histories and data from thousands of other patients to generate personalized insights for clinical decisions and day-to-day challenges. Starting in rare and complex conditions, patients share their data in exchange for value, enabling biopharma and researchers with seamless access to patients and regulatory-grade data, shaving years off drug development for much-needed treatments.Citizen is founded by experienced entrepreneurs with multiple successes under their belts and is funded by top-tier investors including 8VC, Transformation Capital, and Headline Ventures, among others. We are a mission-driven team excited to be building the future of consumer healthcare.The RoleCitizen Health is seeking AI/Machine Learning Engineers 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.ResponsibilitiesDesign and implement end-to-end machine learning solutions, from data preprocessing to model deployment and monitoringDevelop 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 deploymentImplement ML systems that can effectively process and analyze diverse healthcare data types, including structured clinical data, medical imaging, and unstructured textCollaborate with backend engineers to integrate ML models into our production infrastructureEnsure ML systems meet strict healthcare compliance requirements while maintaining high performanceDesign and implement monitoring systems for model performance, data drift, and system healthLead research initiatives to explore and implement state-of-the-art AI/ML techniquesCollaborate with clinical experts to validate model outputs and ensure medical accuracyStay current with the latest developments in AI/ML, particularly in healthcare applicationsWho You AreYou are a builder and problem solver at heart who loves working in fast-paced, innovative environments. You have experience working in consumer technology companies and have also worked on products that deal with complex data and systems. You love start-ups and can operate autonomously knowing that collaboration is the key to your success.Must-Have Skills5+ years of experience in machine learning engineering, with a focus on production ML systemsStrong proficiency in Python and AI/LLM frameworks (LangChain, LangGraph, AutoGen, CrewAI)Experience designing and implementing production-grade ML pipelinesProven track record of deploying ML/LLM solutions in production environmentsStrong understanding of ML fundamentals, including deep learning, NLP, and statistical modelingExperience with ML ops tools and best practicesProficiency in working with large-scale datasets and distributed computingStrong software engineering skills and coding best practicesDegree in Computer Science or equivalent degreePreferred SkillsHands-on experience integrating and optimizing Large Language Models (LLMs) in production environmentsProficiency with LLM tools and techniques such as LangChain, LlamaIndex, Prompt EngineeringExperience implementing Retrieval-Augmented Generation (RAG) systemsFamiliarity with vector databases and efficient similarity search techniques for AI applicationsExperience with message queuing systems and data streaming platformsFamiliarity with data processing and analytics technologiesProven track record of high-quality coding and architectural designPassion for creating robust, scalable, AI-driven systems that make a real-world impact in healthcareWhy Join Us?At Citizen Health, You Will Be Part Of An Ambitious Mission To Build Innovative AI-driven Solutions That Redefine The Consumer Healthcare Experience And Improve Outcomes For Millions. We Value Curiosity, Creativity, And Collaboration, And We Believe In Empowering Every Team Member To Have An Impact. Here, You’ll FindThe chance to work on pioneering AI technologies and solve complex problems that push the boundaries of what’s possible in healthcareThe freedom to lead your projects and shape the company’s direction in a mission-driven environment where your work makes a real impactA fast-paced, high-growth setting that brings expanding opportunities and clear paths for career progressionA supportive culture of learning, experimentation, and innovation, where new ideas are encouraged and exploredA commitment to empathy and diversity, recognizing that our greatest strengths come from our varied perspectives and experiencesRegular team activities and knowledge sharing, all within a culture that prioritizes well-being and connectionAdditional PerksCompetitive salary + equity packageComprehensive health, dental, and vision insuranceUnlimited paid time off, including a generous parental leaveFlexible hybrid work environmentOur Commitment to Diversity & Inclusion: Citizen Health is proud to be an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. We welcome applicants of all backgrounds, identities, and experiences. Everyone deserves an equal chance to contribute and grow here — regardless of race, gender identity, sexual orientation, religion, national origin, age, disability, or veteran status.Don't meet every qualification? No worries! We believe that passion, curiosity, and the right mindset are just as important as a checklist of skills. If you’re excited about what we’re building, we encourage you to apply.Compensation Range: $150K - $300K