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 Youll Be Doing:
Develop clinical AI models to support Ks 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
Requirements: 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
This position is open to all candidates.