Platform Engineer

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

Benefits offered by this job

In-house gym

Lunch buffet

Rooftop terrace with view

Job summary

RaySearch in Stockholm is seeking a Machine Learning Platform Engineer to help build the full model lifecycle platform, enabling ML teams to curate, train and validate models.

This office-first role focuses on robust APIs, reliable services and scalable data and infrastructure, working with ML engineers, software, infra and medical physics teams to deploy and monitor production models.

Qualifications

• M.Sc. in Computer Science, Software Engineering, ML or equivalent with 5+ years of professional software engineering.
• Strong Python skills and backend/API design experience.
• Solid knowledge of relational databases and data modeling (PostgreSQL).
• Experience with containers (Docker) and running production services.
• Good engineering practices: automated testing, code review, CI/CD and observability.
• Experience contributing to architecture decisions in large, shared codebases.
• Excellent spoken and written English.

Responsibilities

• Develop and maintain the model development platform across data, training, validation and deployment.
• Build scalable APIs and services used by ML engineers and other teams.
• Implement data curation, ETL, validation reporting and data lineage.
• Enable production deployment and inference optimization for clinical models.
• Ensure reliability, observability and operational monitoring of platforms.

Skills

Python

API design

Backend services

Data modeling

CI/CD

Observability

Architecture decisions

English communication

Education

M.Sc. in Computer Science or related

Tools

Flyte

Docker

Kubernetes

PostgreSQL

Git

Job description

RaySearch is expanding its machine learning capabilities and is looking for a Machine Learning Platform Engineer to join our headquarters in Stockholm. This is an office-first role.

You will help build our model development platform: the systems that take clinical data through curation, training and validation to deployed models. Our deep learning models are used in RayStation, our treatment planning system, at cancer clinics worldwide. The platform you build is how the next generation of these models gets made.

This is not primarily a model development role. Your focus is the platform, data and infrastructure that let ML engineers develop, validate and deploy models reliably.

About The Job

You will join RaySearch's Machine Learning department and help build a platform that covers the full model lifecycle, from clinical data coming in to validated models going out.

The Work Spans

• Data curation and management: ingesting, organizing and validating large clinical datasets and medical images, with ETL, databases and storage that keep track of where data came from and how it was used.
• Model training: reproducible training pipelines orchestrated with Flyte, for dose prediction, segmentation, detection, classification and image registration, with monitoring of training runs.
• Validation: automated evaluation and reporting that connects data, method, code and results, including web-based tools for reviewing images, contours, radiotherapy plans and validation outcomes.
• Deployment and ML integration: getting models into production and integrated into our software products, including inference optimization.
• Operations: monitoring training and evaluation runs, keeping services and databases up to date, and troubleshooting when things go wrong.
• The platform is used by multiple teams and projects, so good design matters: well-designed APIs, solid documentation and reliable services. You will own services from design to operation and work closely with ML engineers, software developers, infrastructure teams, medical physicists and product teams.

Your profile

You are an experienced software engineer who enjoys building systems that help others do their best work. You are pragmatic: you know when a simple solution is enough and when the problem needs a more scalable architecture. You care about reliability, automation and developer experience. You like building platforms that are robust and well-designed enough to serve many different users.

You don't need to spend your days training neural networks. You do understand how ML development differs from traditional software engineering, and you find problems like reproducibility, data lineage, model versioning, large datasets and GPU workloads interesting.

Requirements

• M.Sc. in Computer Science, Software Engineering, Machine Learning or equivalent experience, and at least 5 years of professional software engineering.
• Strong Python skills and experience designing backend services and APIs.
• Solid knowledge of relational databases and data modeling, such as PostgreSQL.
• Experience with containers, for example Docker, and with designing, running and troubleshooting production services.
• Good engineering practices: automated testing, code review, CI/CD and observability.
• Experience contributing to architecture decisions in large, shared codebases.
• Excellent spoken and written English.

Meriting

• ML platforms and MLOps tooling, such as MLflow and experiment tracking.
• PyTorch, and model deployment or inference optimization, such as TensorRT.
• Workflow orchestration and ETL. We use Flyte, but experience with Airflow, Prefect, Dagster or similar is just as relevant.
• Kubernetes, cloud platforms and infrastructure as code.
• Designing platforms and APIs used by many teams or users, including access control and security.
• Operating production services with high reliability requirements.
• Web application development.
• Medical imaging (DICOM, RT structures), healthcare technology or other regulated, safety‑critical environments.

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

• 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.