CareerPlanGet AI match score →

Machine Learning Engineer

Stockholm, Sweden💼 Full-time🗓 2026-05-27 → 2026-06-24

Core

Building deep learning models to predict radiation dose distributions for cancer treatment planning, enabling faster and higher-quality plans for clinics worldwide.

Role type

Senior Machine Learning Engineer (Medical Imaging/Radiation Therapy)

Builds

Production deep learning models for 3D dose prediction integrated into RayStation software

Domain

Healthcare / Radiation Therapy / Medical Physics

Deliverable

production ML models

Required skills

Deep learning frameworks (PyTorch, TensorFlow), Python (production-quality code), software engineering practices (CI/CD, testing, version control), statistics and applied mathematics, data structures and algorithms

Preferred skills

Computer vision, Transformer/CNN/U-Net architectures, MLOps tooling, LLM fine-tuning (SFT, LoRA, DPO, RLHF), C#/systems-level languages, regulated/safety-critical software experience

Technologies

PyTorch, TensorFlow, RayStation

Responsibilities

Train, evaluate, and deploy deep learning models for dose prediction; collaborate with medical experts to validate results; develop and maintain production code in RayStation; work across the full lifecycle from research to deployment

Seniority

Senior, hands-on IC

Sourced via jobtech · Listed on CareerPlan, which tracks 70,000+ jobs from 20+ sources.
Apply on JobTech ↗