Midjourney
Posted 1mo ago

Principal Machine Learning Engineer (Reconstruction / Quantitative Imaging)

Midjourney
San Francisco, California, United States
OnsiteFull Time
Responsibilities
  • building models
  • productionizing models
  • defining metrics
Requirements
  • Strong applied ML experience with imaging or signal processing
  • Ability to move between research prototypes and production systems
  • Strong evaluation discipline, and experience applying ML to physics-based/inverse problems
Technical tools mentioned
PiNNsNeural OperatorsAgentic-SciMLGPUKalman filtering

Job description

What you’ll do

  1. Partner with medical image reconstruction scientists / engineers to build ML components that improve reconstruction quality, speed, robustness, or quantitative accuracy.

  2. Define training/evaluation pipelines, datasets, and metrics that map to user needs and design requirements.

  3. Productionize models: inference performance, reproducibility, monitoring for drift/regressions, and safe fallbacks.

  4. Collaborate on hybrid algorithms, incorporating physics and learned priors, denoisers, learned regularizers, and quality estimation.

  5. Help build tooling for rapid experimentation as well as rigorous verification of algorithm changes.

What we’re looking for

  • Strong applied ML experience plus comfort with signal processing / imaging or adjacent domains.

  • Ability to move fluidly between research prototypes and production-quality systems.

  • Strong evaluation discipline: metrics, ablations, data leakage avoidance, and reproducibility.

  • A demonstrated track record of applying ML to physics-based or inverse problems (i.e., shipped projects, a portfolio, or publications.)

Useful experience

  • ML for imaging/inverse problems (or adjacent) with strong evaluation discipline and comfort with GPU performance constraints.

  • Pragmatic production mindset: reproducible training/inference, regression testing, and safe deployment in high-stakes contexts.

  • A background in computational physics or scientific computing.

  • Leverage ML-based methods such as PiNNs and Neural Operators to solve partial differential equations arising in ultrasound simulation and imaging.

  • Experience in Agentic-SciML is a plus.

  • Hands-on experience with data curation for ML: building datasets from messy, real-world sources, defining ground truth, and managing labeling or simulation pipelines.

  • Background in data assimilation: combining observations with physics-based models (Kalman filtering, variational methods, ensemble approaches, or learned variants).

About Midjourney

Independent research lab developing generative AI and medical imaging technology.

Similar jobs

Machine Learning Engineer roles near San Francisco, California
1h
Save
Mark Applied
Hide
Machine Learning Engineer, Ads
San Francisco or North America
$165k-$230k/yr HybridFull Time
Higgsfield
Higgsfield: AI platform for cinematic social media video production.
Deep advertising ML experience, strong ads systems knowledge, hands-on LLM or generative AI experience, prompt engineering, model evaluation, post-training, and production ML engineering skills.
LLMs, prompt engineering, supervised fine-tuning, reinforcement learning, CTR, CVR, ROAS
7h
Save
Mark Applied
Hide
Staff Machine Learning Engineer, Causal Inference
San Francisco or Sunnyvale or Los Angeles or Seattle or New York City
$204k-$299k/yr OnsiteFull Time
DoorDash
DoorDashNASDAQ: DASH: On-demand delivery platform connecting consumers with local merchants.
Deep causal inference, econometrics, experimentation, or causal ML experience; production ML engineering; rigorous model evaluation; and ability to collaborate across engineering, analytics, product, and business teams.
ML, CUPED, Covey, Covey Scout for Inbound
8h
Save
Mark Applied
Hide
Principal Machine Learning Engineer, Geometric Vision
Sunnyvale, California, United States
$407k-$460k/yr HybridFull Time
Wayve
Wayve: Develops end-to-end artificial intelligence for autonomous driving systems.
Principal-level expertise in 3D computer vision or geometric machine learning; large-scale deep learning; Python and C++; mathematical foundations; and experience taking research systems through data, training, evaluation, and deployment.
PyTorch, Python, C++, NeRFs, Gaussian Splatting, SLAM, LiDAR
8h
Save
Mark Applied
Hide
Principal Machine Learning Engineer, Geometric Vision
Sunnyvale, California, United States
$407k-$460k/yr HybridFull Time
Wayve
Wayve: Develops AI software for autonomous vehicle navigation.
Principal-level expertise in 3D or geometric vision, deep learning at scale, Python and C++, mathematical foundations, and leading research problems from concept through production systems.
PyTorch, Python, C++, SLAM, NeRFs, Gaussian Splatting, LiDAR
9h
Save
Mark Applied
Hide
Machine Learning Engineer
Austin or Fremont or Lisle
$170k-$250k/yr RemoteFull Time
Koch Inc.
Koch Inc.: Operates global manufacturing, refining, and industrial technology businesses.
10+ YOE10+ years building production ML for physical or engineering systems; strong Python with PyTorch or TensorFlow; engineering and simulation knowledge; Azure Machine Learning or similar cloud ML experience.
Azure Machine Learning, Python, PyTorch, TensorFlow, Azure ML, Gaussian processes, gradient-boosted trees, GNNs, PINNs, Azure GPU compute, Azure ML endpoints, model registry, EM simulation tools
9h
Save
Mark Applied
Hide
Senior Machine Learning Engineer, Content Platform
San Jose, California, United States
$141k-$361k/yr HybridFull Time
Roku
RokuNASDAQ: ROKU: Operates a TV streaming platform and sells streaming hardware.
10+ YOE10+ years in software or ML engineering; BS or MS in computer science, mathematics, or a related quantitative field; expertise deploying large-scale ML models and using modern ML stacks and MLOps tools.
Generative AI, Vision Models, MLOps, LLMs, PyTorch, TensorFlow, Spark, cloud-native MLOps tools
1d
Save
Mark Applied
Hide
Machine Learning Engineer
South San Francisco, California, United States
$212k-$318k/yr HybridFull Time
Stripe
Stripe: Provides online payment processing and financial infrastructure for businesses.
2+ YOEBachelor's degree in a relevant quantitative or computing field and 2+ years building, deploying, and operating production ML systems, including model development, data pipelines, and large-scale deployment.
PyTorch, TensorFlow, XGBoost, Apache Spark
1d
Save
Mark Applied
Hide
Machine Learning Engineer, Tapestry
Mountain View or United States or United Kingdom or Chile or New Zealand or Australia or Brazil
$166k-$244k/yr HybridFull Time
X, The Moonshot Factory
X, The Moonshot FactoryNASDAQ: GOOGL: Develops early-stage technologies and experimental projects.
Bachelor's or master's degree in ML, computer science, statistics, or related field; ML engineering experience; Python and PyTorch or TensorFlow; expertise in specified ML domains.
Python, PyTorch, TensorFlow, AWS, GCP, Azure