Senior AI Researcher

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

• 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
• Collaborative team member who contributes to shared technical decisions and knowledge sharing
• (Preferred) Experience with medical data or medical AI

Who We Are: 

Behind every leading health system is K Health’s AI-powered virtual care engine. 

Esteemed health systems like Mayo Clinic, Cedars-Sinai, Mass General Brigham, Hackensack Meridian Health, and Hartford Healthcare partner with K Health to build and run modern primary virtual care clinics on their behalf. 

Our deeply integrated model modernizes the primary care loop by using AI to put humans first. For our patients, we offer clinical AI (i.e., PatientGPT) and unparalleled access to close care gaps around the clock. For our Providers, we deliver provider-serving agentic solutions (i.e., Perfect Note) to eliminate administrative overload and burnout. And for the health systems, we deploy our top-grade Virtualists in AI-powered virtual clinics 24/7 to capture the patients' care journeys at step one, retain the journey through the system for longitudinal care, and strengthen profitability.  

We’re founded in 2016, headquartered in New York City, and backed by nearly $400 million from leading investors including Valor Equity Partners, Claure Group, Mangrove Capital Partners, 14W, Notable Capital, Lerer Hippeau, Primary Venture Partners, Comcast Ventures, PICO Venture Partners, Max Ventures, and other strategic healthcare partners.

We offer competitive compensation packages based on industry benchmarks for function, level, and geographic location. Offer amounts are determined by multiple factors such as a candidate's experience and expertise.  

We are proud to be an Equal Opportunity Employer and consider applicants for employment regardless of race, ethnicity, religion, color, national origin, ancestry, disability, medical condition, genetic information, marital status, sex, gender, gender identity, gender expression, sexual orientation, pregnancy, childbirth and breastfeeding, age, citizenship, military or veteran status, or any other class protected by applicable federal, state, and local laws. We’re deeply committed to building teams as diverse as the patients we serve and strive to cultivate an environment where everyone can bring their most authentic self to work. We depend on our differences to make our team stronger, our workplace more dynamic, and our product accessible to all of our users.

We are committed to maintaining the integrity of our hiring process and ensuring a safe environment for all candidates. All communication for job offers from K Health will come from email addresses ending in @khealth.com. K Health will never ask you to provide financial information about yourself during the recruitment process. We will never use personal email accounts or other domains for official correspondence. Our official job postings are only listed on our official website and reputable job boards. Be cautious of job offers from sources other than these platforms.