ML Platform Engineer

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 across both online and in-person primary care settings. K Health's primary care platform engages patients through a personalized investigation, informed by relevant medical history from the EMR, to provide primary care providers with comprehensive insights and automated workflows at the point of care. Coupled with patient insights, K Health’s technology empowers providers to make individualized diagnoses and treatment decisions and deliver high-quality medicine at scale.

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 continue to partner with a growing roster of major health systems (Cedars-Sinai, Hackensack Meridian Health, Hartford HealthCare), enhancing accessibility and quality of care by pairing people with technology. 

K Health is a modern healthcare company that is reinventing the patient, provider, and AI cooperative for a singular, digitally unified, high-quality care delivery mission. Our cross-functional approach integrates deep expertise in clinical care, operations, data science, product, consumer marketing, and engineering into one team that is transforming the experience of patients, providers, and the overall care delivery system. Join us in our mission to deliver smarter, simpler healthcare!

About the role:

We are looking for a motivated and experienced Machine Learning Platform Engineer to join our dynamic team.
In this role, you will collaborate closely with data scientists and DevOps professionals to design and build the infrastructure, ecosystem libraries, and pipelines that power our data science initiatives. You will take ownership of model development, monitoring, and maintenance, working hand-in-hand with data scientists on a daily basis.
If you’re passionate about AI, machine learning, and writing high-quality code—and are eager to contribute to innovative, impactful “do good” projects in the digital health space—we’d love to hear from you!

What you'll be doing:

• Design, develop, and maintain our machine learning ecosystem libraries.
• Build and manage data science code, Docker images, and Kubeflow Pipelines (KFP).
• Create and maintain CI scripts to ensure seamless integration and delivery.
• Conduct thorough code reviews to uphold high-quality standards.
• Collaborate closely with data scientists, understanding and addressing their evolving needs.
• Work alongside software developers to seamlessly integrate machine learning models into production systems.
• Stay current with the latest advancements in machine learning, leveraging innovative techniques to enhance the company’s products and services.

What we're looking for:

• 5+ years of hands-on industry experience with Python or C++/Rust/Java/Go.
• 3+ years of experience in backend or platform development.
• 2+ years of experience working with cloud platforms such as Google Cloud (preferred), Azure, or AWS, including familiarity with ML workflow frameworks like KFP or Vertex Pipelines.
• Proficiency with Docker.
• Solid experience in ML/AI development (a must).
• Expertise with transformers, PyTorch, CUDA, and other low-level ML libraries.
• Familiarity with inference optimization libraries (e.g., TGI, VLLM, etc.).
• Knowledge of tuning frameworks and libraries (e.g., TorchTune, TRL, Unsloth).
• Familiarity with Kubernetes.
• Strong focus on code quality and optimization.
• Excellent problem-solving skills and a proactive attitude.
• Collaborative mindset with the ability to work closely with cross-functional teams. Strong communication and teamwork skills are essential.

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.