Senior Machine Learning Engineer — Virtual Staining & Digital Pathology

Senior Machine Learning Engineer — Virtual Staining & Digital Pathology 📍 Remote · US or Canada · Full-time 💰 $165K–$225K per year 🧠 Machine Learning · Computer Vision · Generative Modeling · Digital Pathology 🛂 Visa sponsorship not available⛔ BEFORE YOU APPLY — these are hard gates, not flexible • You must be a US or Canadian citizen. No visa sponsorship, no H1B transfers, no exceptions. • 5–10 years of experience specifically as a Machine Learning Engineer or ML Inference Engineer. "Software Engineer" or "Full Stack Engineer" with tangential or supporting ML work does not count. • Real, verifiable production experience with TensorRT — not just general familiarity with PyTorch/TensorFlow. • Real experience with large-format imagery (2000×2000px or larger): pathology, satellite, drone, microscopy, or equivalent formats.If you don't meet the 4 points above, please don't apply at this time — this lets us review your profile faster for future roles that are a better fit.🎯 Quick self-check (2 minutes) — before you apply • Can you name the specific TensorRT version you've used, and a real technical problem you solved while converting a model? If your answer is "yes, I've used it" without being able to give a specific example, this role probably isn't the right fit yet. • Can you name the tile/patch size you used in your most recent large-image project, and how you handled boundary reconstruction? • Does your salary expectation fall within $165K–$225K? • If you answered yes to all 3 with confidence, keep reading and apply at the end of this posting.About the company A virtual staining and digital pathology company building AI-powered technology for clinical workflows. We develop machine learning systems that support high-fidelity virtual staining, large-scale pathology image analysis, and production-ready model deployment for real-world healthcare environments.About the role We are seeking an experienced Senior Machine Learning Engineer to own the representation-learning and generative modeling stack powering our virtual staining technology. The ideal candidate has deep expertise in machine learning, computer vision, image processing, generative modeling, and production-ready ML systems — with the ability to build generalizable models and evaluations that can stand up in real clinical workflows. This is a senior role for someone with 5–10 years of experience as a Machine Learning Engineer or ML Inference Engineer, ideally at a top tech company.What you'll do • Design and implement novel computer vision and deep learning algorithms for virtual staining and digital pathology applications. • Conduct rigorous experiments to evaluate algorithm performance, validate research hypotheses, and drive iterative improvements. • Develop and advance ML models leveraging Vision Transformers, Diffusion Models, GANs, and other generative architectures for image-to-image translation tasks. • Apply classical and learned image enhancement, denoising, and semantic segmentation techniques to histopathology imaging challenges. • Explore image representation in latent space for efficient, high-fidelity virtual staining. • Stay current with state-of-the-art research and identify opportunities to apply novel techniques to the product roadmap. • Collaborate with ML Engineering and software teams to translate research prototypes into production-ready systems that meet latency and throughput requirements. • Work with large-scale pathology datasets to train, validate, and fine-tune foundation models and custom architectures. • Partner with software engineers, data scientists, and pathology domain experts to integrate research into production systems. • Contribute to best practices for data engineering, data governance, and data quality across research and production pipelines. • Leverage AI coding and ideation tools to accelerate research velocity and prototype new approaches.Must-haves Hard-gated — no exceptions • 5–10 years of experience as a Machine Learning Engineer or ML Inference Engineer, ideally at a top tech company. • US or Canadian citizenship. • Experience with large-format image processing — counts if it's: • ✅ Medical/pathology imaging (WSI, microscopy) • ✅ Satellite or drone imagery at gigapixel scale • ✅ Any imagery you had to tile due to GPU memory constraints • Doesn't count: • ❌ Scanned documents or high-resolution PDFs • ❌ Product/retail imagery, even if "large" • ❌ High-volume streaming/event processing (Kafka, etc.) without an imaging component • Experience with inference processing and model hosting. • Expert proficiency in Python, PyTorch, and TensorRT, with a specific project you can describe in technical depth. • Undergraduate degree in Computer Science from a top 100 school. • Master's or PhD in Computer Science, Electrical Engineering, or a related field (PhD preferred).Evaluated in depth during the technical interview • Hands-on experience with at least one of: Vision Transformers, Diffusion Models, GANs, sem