ML Platform Engineer

Posted 11 Days Ago
Easy Apply
Be an Early Applicant
Tel Aviv
In-Office
40K-120K Annually
Mid level
Healthtech
The Role
Design and implement machine learning models and pipelines while optimizing their performance. Collaborate with data scientists and developers to integrate ML models into production.
Summary Generated by Built In

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

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 in software engineering with experience in backend/platform roles.
  • 5+ years of experience with Python.
  • Proficiency in another language, such as C++, Rust, Java, or Go, is an advantage.
  • 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.
  • Solid experience in ML/AI development (a must).
  • Experience with inference optimization (vLLM) and fine-tuning (Axolotl/Huggingface).
  • Expertise with transformers, PyTorch, CUDA, and other low-level ML libraries.
  • Familiarity with Docker and Kubernetes.
  • Excellent problem-solving skills and a proactive attitude, with a strong focus on code quality and optimization.
  • Collaborative mindset with the ability to work closely with cross-functional teams. Strong communication and teamwork skills are essential.
Benefits & Perks: 
  • 20 paid vacation days, 18 days sick leave.
  • Hybrid work schedule with team meals and stocked fridges
  • Commuter Benefits
  • Community focused events
  • Pension Plan
  • Stipend per day for food
  • Stock options for every full-time employee
  • Vocational Studies Fund

 


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.

Top Skills

Airflow
AWS
Azure
Cuda
GCP
Kubeflow
Python
Sagemaker
Torch
Transformers
Vertex Pipelines
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: New York, NY
363 Employees
Year Founded: 2016

What We Do

We’re a digital healthcare company on a mission to use the power of shared medical knowledge to get everyone access to better, more affordable healthcare—whenever, wherever they need it, 24/7, right from your phone. Our platform consolidates the knowledge of thousands of doctors and billions of clinical data insights to get people trusted information and treatment. We also offer, through affiliated professional entities, the option to chat with a clinician without insurance for less than a copay. We treat almost all primary care conditions, as well as anxiety and depression, and offer customizable treatment plans and prescriptions remotely.

Similar Jobs

Wix Logo Wix

Data Engineer

Enterprise Web • Information Technology • Software
In-Office
Tel Aviv, ISR
6000 Employees
Easy Apply
In-Office
Tel Aviv, ISR
349 Employees
50K-120K Annually

Aidoc Logo Aidoc

Senior Software Engineer

Artificial Intelligence • Healthtech
In-Office or Remote
Tel Aviv, ISR
478 Employees

Aidoc Logo Aidoc

Principal Software Engineer

Artificial Intelligence • Healthtech
In-Office or Remote
Tel Aviv, ISR
478 Employees

Similar Companies Hiring

Camber Thumbnail
Social Impact • Healthtech • Fintech
New York, NY
53 Employees
Sailor Health Thumbnail
Telehealth • Social Impact • Healthtech
New York City, NY
20 Employees
Granted Thumbnail
Mobile • Insurance • Healthtech • Financial Services • Artificial Intelligence
New York, New York
23 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account