Midjourney Logo

Midjourney

Principal Machine Learning Engineer (Reconstruction / Quantitative Imaging)

Posted Yesterday
In-Office or Remote
6 Locations
Expert/Leader
In-Office or Remote
6 Locations
Expert/Leader
Lead development of ML components for medical image reconstruction: define training/evaluation pipelines and metrics, productionize models (inference performance, monitoring, safe fallbacks), collaborate on hybrid physics-ML algorithms, and build tooling for experimentation and rigorous verification.
The summary above was generated by AI
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).

HQ

Midjourney San Francisco, California, USA Office

San Francisco, California, United States

Similar Jobs

Senior level
AdTech • Cloud • Digital Media • Information Technology • News + Entertainment • App development
Design, implement, and optimize Unreal Engine rendering systems for mobile (shaders, materials, lighting, post-processing) in C++. Profile and debug GPU performance on iOS/Android, collaborate with Art/Tech Art/Engine teams, mentor junior engineers, and ensure scalable, stable visuals across diverse mobile hardware.
Top Skills: AndroidAndroid Gpu InspectorC++GitiOSPerforceShadersUnreal EngineUnreal Engine Rendering PipelineXcode Instruments
Senior level
AdTech • Cloud • Digital Media • Information Technology • News + Entertainment • App development
Design, implement, and maintain Unreal Engine rendering systems and shaders. Optimize performance and memory across platforms, debug GPU and lighting issues, participate in profiling, support UE upgrades, collaborate with Art/Tech Art, mentor junior engineers, and contribute to rendering architecture and technical decisions.
Top Skills: C++GitGpu Profiling ToolsGraphics ApisPerforceShadersUnreal Engine
Senior level
AdTech • Cloud • Digital Media • Information Technology • News + Entertainment • App development
Design, implement, and optimize Unreal Engine engine-level systems for mobile (iOS/Android). Focus on performance, memory, power, platform integrations, profiling and debugging on real devices. Collaborate with rendering and gameplay teams, maintain engine code, support mobile features (asset streaming, memory budgeting), mentor junior engineers, and follow team standards for clean Unreal C++ code and version control.
Top Skills: AndroidC++GitiOSPerforceUnreal Engine

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account