Senior AI Researcher

Who we are: Founded in 2016, K Health's mission is to deliver accessible, high-quality healthcare at scale. As a leading clinical AI company in primary care, K Health has developed a suite of clinical AI agents that enhance provider efficiency and improve patient outcomes. K Health's virtual primary care platform is enhanced by an AI copilot to complete the initial patient intake, summarizing relevant history from the EMR, and generating “the perfect note” to reduce the time providers spend on basic data collection and non-clinical tasks. Unlike other virtual medicine companies, K Health delivers comprehensive, longitudinal primary care in a virtual setting, enabling our clinicians to be true primary care providers without sacrificing scope of practice or continuity of care. As a venture-backed startup trusted with nearly $400 million in funding and a $900 million valuation as of July 2024, K Health is well-positioned for sustained future growth. We are expanding our partnerships with major health systems (Cedars-Sinai, Hackensack Meridian Health, Hartford HealthCare), enhancing accessibility and quality of care by pairing people with technology. Our providers are credentialed at these renowned academic health systems and adhere to their clinical guidelines, ensuring patient-first, evidence-based care. Fully integrated into Epic, our AI platform enables K Health to operate as an extension of existing healthcare systems, benefiting from a built-in network of specialists and social services to seamlessly serve patients within those communities. Our unique care delivery model enables our providers to deliver the same high-quality, accessible care to a mixed-payer population, including those with commercial insurance, Medicare, and Medicaid. Join us in our mission to deliver smarter, simpler healthcare of the future - today! About the Role: Our data science team is seeking a highly skilled AI researcher with expertise in modern AI methods (e.g., multi-agent architectures, reinforcement learning on LLMs) and a passion for transforming theoretical challenges into real-world products. In this role, you will leverage our unique clinical datasets to design and build groundbreaking solutions that provide medical benefits to millions of people. You will take full end-to-end ownership of your models - from ideation and research to production deployment and ongoing monitoring. Working across both research and production environments, you should have proven experience in developing and deploying machine learning algorithms, along with strong interpersonal and collaboration skills. We value independent thinkers who are also proactive doers. What You’ll Be Doing: Develop clinical AI models to support K’s clinic workflows, with a focus on patient-facing applications Design and implement scalable, efficient pipelines for data preprocessing, information extraction, and model training Stay up to date with the latest AI research and best practices, and translate them into production-ready solutions Communicate complex technical concepts and solutions to non-technical stakeholders Collaborate with MLE and AI engineering teams to deploy scalable, robust solutions What We’re Looking For 5+ years of experience in AI research, with a strong background in LLM research Advanced proficiency in Python and hands-on experience with leading frameworks and libraries, such as PyTorch, Hugging Face, torchtune, and vLLM M.Sc. or Ph.D. in Computer Science, Data Science, Statistics, Engineering, Mathematics, Physics, or a related field. Preferably with applied focus on machine learning, computer vision, NLP, or deep learning Fast learner with excellent problem-solving skills Positive attitude, intellectual curiosity, and eagerness to learn and share knowledge Passion for medicine, health, and wellbeing, with a drive to make a meaningful impact Proven experience translating deep learning research into reliable, high-impact production systems ... (truncated, view full listing at source)