Software Engineer, AI Engineer, Machine Learning​/ ML Engineer

Job Description Summary

The Caption Health SBU focuses on clinical applications that aid in early disease detection, using AI to assist in conducting ultrasound scans. The Software Engineer helps port machine learning models into our cross-platform SDK. This role bridges the gap between Python-based model development and C++ SDK integration, ensuring high-performance inference across iOS, Android, Linux, and Windows platforms.

We cannot sponsor for this role, no OPT

• Port ML models (primarily PyTorch and Tensor Flow, exported to ONNX) into our C++ SDK.
• Translate and optimize Python-based postprocessing logic into performant C++ code.
• Collaborate with ML scientists to understand model behavior and ensure fidelity in deployment.
• Optimize inference performance for mobile and desktop platforms, considering memory, latency, and compute constraints.
• Maintain clean, modular C++ APIs that expose ML functionality to client applications.
• Write unit and integration tests to validate model outputs and SDK stability.
• Participate in architectural discussions and provide technical guidance to help the team make well-informed decisions regarding feature implementation.
• Write clean, maintainable, and well-documented code, with an emphasis on scalability and performance.
• Other duties as assigned.

• Bachelor’s degree or equivalent practical experience in Computer Science, Engineering, or a related technical field.
• 4+ years of experience working as a software engineer, with a proven track record of working with software development kits (SDKs) and client-side applications.
• 4+ years of experience in C++ development, with a focus on SDKs or embedded systems.
• 2+ years of strong fundamentals in statistics.
• Proficiency in Python, especially for understanding and translating ML pipelines.
• Familiarity with mobile development constraints (iOS/Android) and cross-platform build systems.
• Strong debugging and performance profiling skills.
• Experience with AI/ML integration or development within software systems is highly preferred.

• Experience with model quantization, pruning, or hardware acceleration (e.g., Metal, NNAPI).
• Experience deploying ML models using ONNX Runtime, Tensor Flow Lite, or similar frameworks.
• Familiarity with medical imaging or real-time signal processing.
• Exposure to CI/CD pipelines for ML model validation and SDK packaging.

We will not sponsor individuals for employment visas, now or in the future, for this job opening. For U.S. based positions only, the pay range for this position is $-$ Annual. It is not typical for an individual to be hired at or near the top of the pay range and compensation decisions are dependent on the facts and circumstances of each case.

The specific compensation offered to a candidate may be influenced by a variety of factors including skills, qualifications, experience and location. In addition, this position may also be eligible to earn performance based incentive compensation, which may include cash bonus(es) and/or long term incentives (LTI). GE Health Care offers a competitive benefits package, including not but limited to medical, dental, vision, paid time off, a 401(k) plan with employee and company contribution opportunities, life, disability, and accident insurance, and tuition reimbursement.

Relocation Assistance Provided: No

GE Health Care offers a great work environment, professional development, challenging careers, and competitive compensation. GE Health Care is an Equal Opportunity Employer. Employment decisions are made without regard to race, color, religion, national or ethnic origin, sex, sexual orientation, gender identity or expression, age, disability, protected veteran status or other characteristics protected by law.

GE Health Care will only employ those who are legally authorized to work in the United States for this opening. Any offer of employment is conditioned upon the successful completion of a drug screen (as applicable).

While GE Health Care does not currently require U.S. employees to be vaccinated against COVID-19, some GE Health Care customers have vaccination mandates that may apply to certain GE Health Care employees.

Seniority level: Mid-Senior level

Employment type: Full-time

Job function: Engineering and Information Technology

Industries: Hospitals and Health Care