Staff Machine Learning Engineer - AI Imaging

Discover International; United States; Hybrid; USD 140,000 - 190,000; Full time

Benefits offered by this job

Equity

Flexible work options

Hybrid work environment

Job summary

Discover International is seeking a Staff Level Machine Learning Engineer to lead design, deployment, and optimization of production-grade ML systems in healthcare AI.

The role focuses on scalable pipelines, real-time inference, and cross-functional collaboration with data scientists and product teams. Hybrid work across major U.S. tech hubs is available and equity is offered.

Qualifications

• Experience building and deploying ML models in production environments.
• Strong programming background in Python, Go, C++, or Java.
• Experience with distributed training, data pipelines, and real-time inference systems.
• Solid understanding of ML concepts, particularly in computer vision or LLMs.
• Deep expertise with cloud platforms (AWS/GCP/Azure) and DevOps (Kubernetes, Terraform, Docker).

Responsibilities

• Collaborate cross-functionally with data scientists, engineers, and product teams to design and deliver scalable ML solutions.
• Mentor and guide other engineers, setting standards and shaping ML engineering practices.
• Architect robust systems that support high-volume batch and real-time inference.
• Develop APIs, feature stores, and automation tooling to accelerate ML experimentation and deployment.
• Build and maintain pipelines for data ingestion, training, evaluation, and model serving.
• Implement monitoring, observability, and reliability frameworks for production-grade performance.
• Optimise performance across the ML stack for latency, scalability, and cost-efficiency.

Skills

Python

Go

C++

Java

Distributed training

Data pipelines

Real-time inference

ML concepts

Computer vision

LLMs

Cloud platforms

DevOps

Job description

We're looking for a Staff Level Machine Learning Engineer build large-scale systems that make a real-world impact?

Our client — a fast-growing AI company revolutionising healthcare — is seeking a Staff Machine Learning Engineer to lead the design, deployment, and optimisation of production-grade ML systems.

This is your opportunity to shape the next generation of applied AI in healthcare and work with some of the brightest minds in the industry.

Responsibilities:

• Collaborate cross-functionally with data scientists, engineers, and product teams to design and deliver scalable ML solutions.
• Mentor and guide other engineers, setting standards and shaping the company’s ML engineering practices.
• Architect robust systems that support high-volume batch and real-time inference.
• Develop APIs, feature stores, and automation tooling to accelerate ML experimentation and deployment.
• Build and maintain pipelines for data ingestion, training, evaluation, and model serving.
• Implement monitoring, observability, and reliability frameworks to ensure production-grade performance.
• Optimise performance across the ML stack — balancing latency, scalability, and cost-efficiency.

Requirements:

• Experience building and deploying ML models in production environments.
• Strong programming background in Python, Go, C++, or Java.
• Experience with distributed training, data pipelines, and real-time inference systems.
• Sound understanding of ML concepts, particularly in computer vision or large language models (LLMs).
• Deep expertise with cloud platforms (AWS, GCP, or Azure) and DevOps infrastructure (Kubernetes, Terraform, Docker).

Why apply?

• Competitive compensation and equity package.
• Flexible and hybrid work options across major U.S. tech hubs.
• A high-growth, collaborative culture where ownership and innovation are valued.
• A mission-driven environment — shaping the future of AI-powered healthcare.