Research Engineer - ML Infrastructure

Reposted 3 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

Skills Required

  • 3+ years building ML infrastructure
  • Strong Python skills
  • Expertise in PyTorch or JAX
  • Hands-on experience with data pipeline technologies
  • Experience with distributed systems and cloud infrastructure
  • Track record of building scalable data systems
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

Applied Systems Logo Applied Systems

Sr. Director, NA People Business Partners

Cloud • Insurance • Payments • Software • Business Intelligence • App development • Big Data Analytics
Remote or Hybrid
United States
3040 Employees
170K-200K Annually

CoreWeave Logo CoreWeave

Counsel

Cloud • Information Technology • Machine Learning
In-Office
5 Locations
1450 Employees
177K-237K Annually

Airwallex Logo Airwallex

Engineering Manager

Artificial Intelligence • Fintech • Payments • Business Intelligence • Financial Services • Generative AI
Remote or Hybrid
San Francisco, CA, USA
2000 Employees

Similar Companies Hiring

Fairly Even Thumbnail
Hardware • Other • Robotics • Sales • Software • Hospitality
New York, NY
30 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
Onshore Thumbnail
Software
US
100 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account