Research Engineer - ML Infrastructure

Reposted 13 Days Ago
San Francisco, CA, USA
In-Office
Mid level
Healthtech • Software
The Role
We're looking for a Research Engineer to develop ML infrastructure and data systems for medical imaging, focusing on building data pipelines and model deployment.
Summary Generated by Built In
About Us

We're tackling one of healthcare's most critical challenges in medical imaging and diagnostics. Our company operates at the intersection of cutting-edge AI and clinical practice, building technology that directly impacts patient outcomes. We've assembled one of the industry's most comprehensive and diverse medical imaging datasets and have a proven product-market fit with a substantial customer pipeline already in place.

Role Overview

We're seeking a research engineer to bridge the gap between research and production, building ML infrastructure and data systems for medical imaging at scale. You'll own critical data pipelines that unify live production traffic with offline datasets, design storage solutions for multimodal medical data, and build training + inference infrastructure that enables our research team to iterate rapidly. This role requires someone who can move fluidly between model training, data engineering, ML systems, and production deployment.

Key Responsibilities
  • Build and optimize distributed ML infrastructure for training foundation models on large-scale medical imaging datasets.

  • Design and implement robust data pipelines to collect, process, and store large-scale multimodal medical imaging data from both production traffic and offline sources.

  • Build centralized data storage solutions with standardized formats (e.g., protobufs) that enable efficient retrieval and training across the organization.

  • Create model inference pipelines and evaluation frameworks that work seamlessly across research experimentation and production deployment.

  • Collaborate with researchers to rapidly prototype new ideas and translate them into production-ready code.

  • Own end-to-end delivery of ML systems from experimentation through deployment and monitoring.

Qualifications
  • 5+ years building ML infrastructure, data pipelines, or ML systems in production

  • Strong Python skills and expertise in PyTorch or JAX

  • Hands-on experience with data pipeline technologies (e.g., Spark, Airflow, BigQuery, Snowflake, Databricks, Chalk) and schema design

  • Experience with distributed systems, cloud infrastructure (AWS/GCP), and containerization (Docker/Kubernetes)

  • Track record of building scalable data systems and shipping production ML infrastructure

  • Ability to move quickly and handle competing priorities in a fast-paced environment

Preferred Qualifications
  • Experience with reinforcement learning training pipelines (e.g., RLHF, reward modeling, or online learning systems)

  • Support A/B testing and experimentation workflows for model rollouts, including monitoring statistical significance and managing canary deployments.

  • Familiarity with vision-language models (VLMs) or multimodal architectures

  • Experience with medical imaging formats (DICOM) and healthcare data standards

  • Background in distributed training frameworks (PyTorch Lightning, DeepSpeed, Accelerate)

  • Familiarity with MLOps practices and model deployment pipelines

  • Experience with privacy-preserving data systems and HIPAA compliance

Top Skills

Airflow
BigQuery
Databricks
Docker
Jax
Kubernetes
Python
PyTorch
Snowflake
Spark
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
6 Employees
Year Founded: 2024

What We Do

Epsilon Health exists to solve the looming global radiology crisis before it reshapes patient care. Radiology underpins nearly every medical specialty, yet it is one of the most strained parts of the healthcare system. We’re rethinking how imaging and interpretation work to make radiology faster, more reliable, and future-proof.

Similar Jobs

Third Dimension AI Logo Third Dimension AI

Research Engineer - Data Infrastructure/ML

Artificial Intelligence • Information Technology • Software
Remote or Hybrid
9 Locations
16 Employees

Boeing Logo Boeing

Component Maintenance Engineer (Associate or Experienced)

Aerospace • Information Technology • Software • Cybersecurity • Design • Defense • Manufacturing
In-Office
Seal Beach, CA, USA
170000 Employees
91K-151K Annually

Boeing Logo Boeing

Devops Engineer

Aerospace • Information Technology • Software • Cybersecurity • Design • Defense • Manufacturing
In-Office
Seal Beach, CA, USA
170000 Employees
99K-133K Annually

Boeing Logo Boeing

Software Engineer

Aerospace • Information Technology • Software • Cybersecurity • Design • Defense • Manufacturing
In-Office
Huntington Beach, CA, USA
170000 Employees
106K-144K Annually

Similar Companies Hiring

Milestone Systems Thumbnail
Software • Security • Other • Big Data Analytics • Artificial Intelligence • Analytics
Lake Oswego, OR
1500 Employees
Fairly Even Thumbnail
Software • Sales • Robotics • Other • Hospitality • Hardware
New York, NY
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account