Machine Learning Engineer

RaySearch Laboratories; Stockholms kommun; On-site; SEK 700,000 - 900,000; Full time

Benefits offered by this job

In-house gym

Yoga classes

Lunch buffet

Daily fika

Rooftop terrace

Job summary

RaySearch Laboratories in Stockholm seeks a Machine Learning Engineer with strong software engineering skills to join our head office. You will develop ML solutions for radiation oncology, train and deploy deep learning models, and collaborate with medical physicists and oncologists to ensure clinical-grade performance.

You will work on 2D/3D medical imaging, segmentation, synthesis, and dose prediction, building scalable pipelines and contributing to shared architecture decisions.

Qualifications

• MSc in CS/ML or equivalent required.
• 4+ years in software development or ML.
• Strong Python and production‑quality coding.
• Experience with PyTorch and ML frameworks.
• Excellent English communication skills.

Responsibilities

• Train, evaluate, and operationalize deep learning models.
• Analyze results with clinical metrics and ensure acceptance criteria.
• Work with real clinical datasets and validate with clinicians.
• Collaborate with medical physicists, oncologists, and software engineers.
• Design experiments and evaluate new architectures and training strategies.
• Build reliable pipelines for training, validation, evaluation, and inference.

Skills

Python

ML techniques

Software engineering

CI/CD

English

Education

M.Sc. in Computer Science or ML

Tools

PyTorch

Job description

RaySearch is expanding in the field of machine learning and is looking for a Machine Learning Engineer with strong software engineering skills to join our head office in Stockholm. In this role, you will have the opportunity to innovate and develop software that improves the lives of cancer patients around the world, while building the infrastructure, services, and developer tooling that enable machine learning solutions in the ML department to move efficiently from experimentation to clinical products used in cancer treatment worldwide.

About The Job

You will be part of an agile team developing machine learning solutions for radiation oncology. Our work spans areas such as image segmentation, image synthesis, treatment planning, dose prediction, and other deep learning applications within medical imaging and radiation therapy.

You will

Your day-to-day work will involve training, evaluating and operationalizing deep learning models. This includes analyzing results using clinical metrics and ensuring that predictions meet acceptance criteria. You will work with real clinical datasets and iterate closely with medical physicists and oncologists to validate model performance against clinical standards.

The role combines machine learning research, experimentation and software development

• Develop state-of-the-art deep learning methods for medical imaging and radiation therapy.
• Work on problems involving 2D and 3D medical images, anatomical structures, treatment data, and multimodal clinical information
• Design experiments, define appropriate evaluation methods, and analyze model behavior.
• Build and evaluate new architectures, training strategies, and research ideas and determine which approaches are worth bringing into our products
• Work with large clinical datasets and develop reliable pipelines for training, validation, evaluation, and inference
• Collaborate closely with medical physicists, oncologists, researchers, product owners, and software engineers

Your profile

You are someone who genuinely loves both ML and software engineering. You read papers, get excited about new architectures and training techniques, and have expertise in implementing them.

You are autonomous and self-driven. Given a problem, you take ownership: you research the landscape, form a plan, and execute without needing to be guided through every step. You know when to dig deeper on your own and when to pull in a colleague. You are comfortable making decisions, and you communicate clearly about what you are doing and why.

You contribute to a positive team atmosphere and take responsibility for your work. Above all, you care about building software that makes a meaningful impact for healthcare professionals and their patients.

Experience & Skills

Requierments

• M.Sc. in Computer Science, Machine Learning, or equivalent
• Minimum 4 years of professional experience in software development, machine learning, or a related field
• Excellent understanding of machine learning techniques and algorithms
• Strong programming skills in Python with emphasis on writing clean, maintainable, production-quality code
• Solid software engineering practices: version control, code review, testing, CI/CD
• Experience using deep learning frameworks such as PyTorch
• Solid knowledge of data structures, algorithms, statistics, and applied mathematics
• Comfort working in large, collaborative code bases and contributing to shared architecture decisions
• Excellent spoken and written communication skills in English

Meriting

• Experience with medical image segmentation, registration, image synthesis, reconstruction, representation learning, or multimodal imaging.
• Contributions to open source projects
• Experience working in regulated or safety-critical software environments (e.g., medical devices, automotive)

Our Culture

At RaySearch, we share a passion for innovation and the fight against cancer. Our team consists of dedicated experts who strive to deliver exceptional results through collaboration, attention to detail, and cutting-edge technology. We take pride in our role as a leader in cancer treatment, developing solutions that truly make a difference for patients worldwide.

Our Offer

We offer a dynamic and inclusive work environment in Hagastaden, Stockholm’s Life Science Hub. Our modern office space includes an in-house gym, yoga classes, and social activities such as ping pong, table football, and after-work events. We also provide a fantastic lunch buffet, daily fika, and a stunning rooftop terrace with a 360-degree view of Stockholm. This comes with a competitive compensation and benefits package.