Arbetsbeskrivning

RaySearch develops innovative software solutions to improve cancer care. About 1000 clinics in more than 40 countries use RaySearch software to improve treatments and quality of life for patients. RaySearch was founded in 2000 and is listed on Nasdaq Stockholm. The headquarters is located in Stockholm, with subsidiaries in the US, Europe and Asia - Pacific. Today we are more than 400 employees with a common vision of improving cancer care with innovative software. Our great staff is crucial for our success and we offer a fantastic working environment in modern offices, flexibility and good opportunities for development. We believe in equal opportunities, value diversity and work actively to prevent discrimination.

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. Joining our team provides you with a unique opportunity to innovate and develop software that improves the lives of cancer patients around the world.

About the job

You will be part of an agile team developing machine learning solutions to improve cancer treatment. As part of the ML Planning team, you will build deep learning models to predict radiation dose distributions used in clinics worldwide.

In radiation therapy, treatment planning involves delivering precise radiation to tumors while sparing healthy tissue—a complex optimization problem. Our models predict 3D dose directly from patient data, enabling faster, high-quality treatment plans and significantly reducing the time and expertise required.

You will

Your work will focus on training, evaluating, and deploying deep learning models for dose prediction, using clinical metrics such as DVH to ensure performance meets medical standards.

You will work with real clinical data and collaborate closely with medical experts to validate results. At the same time, you will develop and maintain production code in RayStation, where engineering quality is as important as model performance.

You take ownership across the full lifecycle—from research to deployment—and work closely with cross-functional teams to build solutions that make a real impact.

Your profile

You genuinely enjoy both research and coding—you read papers for fun and are eager to turn new ideas into practical solutions. You are not only an ML practitioner but also a strong software developer who values code quality, scalable design, and maintainability.

You are autonomous and take ownership: you explore problems, form a plan, and execute while knowing when to collaborate. You communicate clearly about your decisions and progress.

You deliver high-quality results, thrive in fast-paced environments, and contribute positively to the team. Most importantly, you are motivated by building software that makes a real impact in healthcare.

Your profile

You enjoy both research and coding, staying up to date with new ideas and quickly turning them into practical solutions. You are a strong software developer who values code quality, scalable design, and maintainability.

You are autonomous and take ownership—exploring problems, forming a plan, and executing while knowing when to collaborate. You communicate clearly about your work.

You deliver high-quality results, thrive in fast-paced environments, and are motivated by building software that makes a real impact in healthcare.

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 awareness
• Experience using deep learning frameworks such as PyTorch or TensorFlow
• 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 using deep learning for computer vision
• Experience in model development: Transformers, CNNs, U-Nets, or relevant architectures for dose prediction
• Experience with MLOps tooling (experiment tracking, model registry, deployment pipelines)
• Experience fine-tuning and post-training LLMs using Full SFT / LoRA, DPO, RLHF, with custom data curation and evaluation pipelines
• Contributions to open source projects
• Familiarity with C# and/or systems-level languages
• Experience working in regulated or safety-critical software environments (e.g., medical devices, automotive)
• Interest in or exposure to radiation therapy or medical physics

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. 

Application 

Please apply for the position through the application form below. Selection and interviews will be ongoing. We do not accept applications via e-mail.