Machine Learning Engineer

HealthEdge; United States; Hybrid; USD 130,000 - 165,000; Full time

Job summary

HealthEdge is seeking a Machine Learning Engineer to design, build, and ship AI agents and automation across our engineering, product, and delivery teams to solve real-world problems. You will partner with stakeholders in Engineering, Product, and healthcare to understand workflows, identify high-leverage opportunities, and deliver end-to-end AI solutions at scale.

Requirements: 2–4 years of production ML experience, strong Python, experience with LLM APIs and agentic frameworks, and solid

Qualifications

• Master's degree in Computer Science, ML, Data Science, or related field; Bachelor's with relevant experience considered.
• 2–4 years of production ML/AI experience; client-facing or consulting-style roles are a plus.
• Strong Python proficiency; experience with LLM APIs and agentic frameworks; solid ML and software engineering basics.
• Familiarity with healthcare data, EHRs, and regulatory considerations is a plus.

Responsibilities

• Develop and implement AI agents and automation across engineering and delivery processes.
• Collaborate with engineers to integrate AI into existing infrastructure for reliable performance.
• Work with product managers and healthcare teams to identify pain points and deliver measurable outcomes.
• Stay current on LLMs and agentic AI; contribute ideas for innovation and improvement.
• Optimize AI systems for accuracy, latency, cost, and safety with human-in-the-loop guardrails.

Skills

Python

LLM APIs

LangChain

Agentic frameworks

Prompt engineering

PyTorch

scikit-learn

CI/CD

Testing

Version control

Education

Master's degree in CS/ML/DS

Bachelor's with relevant experience

Tools

Git

CI/CD tooling

Docker

Job description

Overview

Machine Learning Engineer, AI Platform

As a Machine Learning Engineer, you will design, build, and ship AI agents and automation that solve real problems across HealthEdge's engineering, product, and delivery organizations, including customer-facing operations. You'll partner directly with stakeholders across Engineering, Product, and healthcare professionals to understand their workflows, identify high-leverage opportunities, and deliver working solutions end-to‑to‑end. Your growing expertise in machine learning and agentic AI will have a direct impact on how HealthEdge builds software, delivers for customers, and operates at scale.

Key Responsibilities

1. AI Platform Development: Develop and implement AI Agents and automation that accelerates internal engineering workflows and customer facing delivery processes, owning the full lifecycle from problem discovery, through prototyping, evaluation, hardening, and production deployment. Contribute reusable libraries, prompt templates, tool-use patterns, and evaluation scaffolding back to the AI Platform.
2. Integration: Partner with software engineers to integrate AI into the company's existing software infrastructure, supporting seamless functionality and performance.
3. Collaboration: Work directly with product managers, implementation consultants, engineers, and business operations teams to identify pain points, scope solutions, and iterate toward measurable outcomes. You are the bridge between what AI can do and what the business needs done.
4. Research and Learning: Stay current with advancements in LLMs, agentic frameworks, machine learning, and healthcare technology, and apply new knowledge to contribute ideas for innovation within the team.
5. Performance and Reliability: Optimize AI systems for accuracy, latency, cost, and safety, with particular attention to human-in-the-loop design and guardrails appropriate for healthcare.
6. Documentation: Maintain clear documentation of model development processes, methodologies, and results to ensure transparency and reproducibility.

Required Qualifications

• Education: Master's degree in Computer Science, Machine Learning, Data Science, or a related field. A Bachelor's degree with relevant experience will also be considered.
• Experience: 2–4 years of experience building and deploying ML or AI systems in production. Experience working directly with non-technical stakeholders or in embedded/consulting-style engineering roles is a strong plus.
• Technical Skills: Strong proficiency in Python. Experience with LLM APIs, agentic frameworks (LangChain, Strands, etc.), and prompt engineering alongside traditional ML frameworks (PyTorch, scikit-learn, etc.). Solid software engineering fundamentals — version control, testing, CI/CD, and comfort operating across the full development lifecycle.
• Healthcare Knowledge: Interest in or familiarity with healthcare data, clinical workflows, and regulatory requirements. Experience working with electronic health records (EHR) or other healthcare datasets is a plus but not.
• Analytical Skills: Strong problem-solving skills and the ability to work with complex datasets to derive actionable insights.
• Communication: Excellent verbal and written communication skills, with the ability to explain technical concepts to non-technical stakeholders.
• Builder Mindset: Energized by turning ideas into working solutions. You balance speed with quality, thrive in ambiguous problem spaces, and pick up new domains quickly.
• Team Player: Ability to work collaboratively in a cross-functional team environment, accept feedback, and contribute to the success of the team.

HealthEdge commits to building an environment and culture that supports the diverse representation of our teams. We aspire to have an inclusive workplace. We aspire to be a place where all employees have the opportunity to belong, make an impact and deliver excellent software and services to our customers.

Geographic Responsibility: While HealthEdge is located in Boston, MA you may live anywhere in the US

*Note this role requires proof of US residency for the last consecutive 5 years*

Type of Employment: Full-time, permanent

Work Environment: The physical demands described here are representative of those that must be met by an employee to successfully perform the essential functions of this job:

• The employee is occasionally required to move around the office. Specific vision abilities required by this job include close vision, color vision, peripheral vision, depth perception, and ability to adjust focus.
• Work across multiple time zones in a hybrid or remote work environment.
• Long periods of time sitting and/or standing in front of a computer using video technology.
• May require travel dependent on company needs.

The above statements are intended to describe the general nature and level of the job being performed by the individual(s) assigned to this position. They are not intended to be an exhaustive list of all duties, responsibilities, and skills. HealthEdge reserves the right to modify, add, or remove duties and to assign other duties as necessary. In addition, reasonable accommodations may be made to enable individuals with disabilities to perform the essential functions of this position in compliance with the Americans with Disabilities Act of 1990. Candidates may be required to go through a pre-employment criminal background check.

HealthEdge is an equal opportunity employer. We are committed to workforce diversity and actively encourage all qualified persons to seek employment with us, including, but not limited to, racial and ethnic minorities, women, veterans and persons with disabilities.

#LI-Remote

** The annual US base salary range for this position is $130,000 to $165,000. This salary range may cover multiple career levels at HealthEdge. Final compensation will be determined during the interview process and is based on a combination of factors including, but not limited to,your skills, experience, qualifications and education.