[Remote] Machine Learning Engineer

Note: The job is a remote job and is open to candidates in USA. HealthEdge provides AI-powered operational infrastructure for health insurance companies and is expanding its healthcare technology capabilities. The Machine Learning Engineer will design, build, and deploy AI agents and automation, collaborate with engineering, product, delivery, and healthcare stakeholders, and optimize reliable AI solutions for healthcare workflows.

Responsibilities

• 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
• Integration: Partner with software engineers to integrate AI into the company's existing software infrastructure, supporting seamless functionality and performance
• 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
• 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
• 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
• Documentation: Maintain clear documentation of model development processes, methodologies, and results to ensure transparency and reproducibility

Skills

• 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
• 2–4 years of experience building and deploying ML or AI systems in production
• 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
• Interest in or familiarity with healthcare data, clinical workflows, and regulatory requirements
• Strong problem-solving skills and the ability to work with complex datasets to derive actionable insights
• Excellent verbal and written communication skills, with the ability to explain technical concepts to non-technical stakeholders
• Energized by turning ideas into working solutions
• You balance speed with quality, thrive in ambiguous problem spaces, and pick up new domains quickly
• Ability to work collaboratively in a cross-functional team environment, accept feedback, and contribute to the success of the team
• Experience working directly with non-technical stakeholders or in embedded/consulting-style engineering roles is a strong plus
• Experience working with electronic health records (EHR) or other healthcare datasets is a plus but not required

Benefits

• Work in a hybrid or remote work environment

Company Overview

Company H1B Sponsorship