Machine Learning Engineer (m/w/d)

Your mission

As a Machine Learning Engineer, you will play a key role in advancing our scientific research and product development efforts. You will contribute to the development of state-of-the-art AI models that power our growing product portfolio, as well as develop robust, scalable deployment pipelines to seamlessly integrate these models into our software systems. Additionally, you will stay up to date on the latest advancements in the field of biomedical AI, actively exploring emerging technologies and identifying novel applications to enhance our capabilities.

Job Key Responsibilities

Develop, train, and deploy deep learning models for histopathological image analysis (e.g., classification, segmentation, instance detection).
Optimize ML models for performance and scalability in production.
Work closely with cloud engineers to ensure seamless deployment on cloud infrastructure.
Collaborate with quality managers to translate R&D models into production-ready, regulatory-compliant tools.
Write and maintain technical documentation for machine learning workflows and systems.
Stay up to date with the latest research in computer vision, deep learning, and biomedical AI.

Your profile

Requirements

Strong proficiency in Python and experience with Deep Learning frameworks (e.g., PyTorch)
Expertise in machine learning workflows, including data preprocessing and model optimization, as well as agentic AI systems.
Hands-on experience working with state-of-the-art computer vision and deep learning models, such as CNNs and Transformers.
Experience working with healthcare data such as histopathology and radiology scans.
Master’s or PhD in Computer Science, Machine Learning, Computational Pathology, or related field.
Ability to work collaboratively in cross-functional teams, including scientists, statisticians, and regulatory experts.
Strong problem-solving, communication, and presentation skills.

Nice to have

Experience working in regulated software development lifecycles (IEC 62304, ISO 13485)
Background in digital pathology.
Familiarity with cloud platforms like AWS for scalable model training and deployment.
Startup or high-growth team experience.
Familiarity with tools like GitHub Copilot, Cursor, Replit, or other AI productivity tools.
Solid engineering practices: version control, testing, containerization.

Why us?

The opportunity to make a real difference for cancer patients.
Competitive salary, stock options, and extra benefits.
A beautiful office near Alexanderplatz with an on-site gym, private showers, free drinks and snacks, rooftop bar, co-working space, and a quiet room for meditation or prayer.
Competing at the global top-tier of cancer research together with the founders, who are renowned researchers in the field of AI for biomarker development from medical imaging.
An extremely fast-paced and high-growth VC-funded start-up environment with key ownership from day 1.

About us

StratifAI is an innovative Berlin-based precision oncology startup developing the next generation of AI-based cancer biomarkers. The founders have published over 100 articles in top-tier journals and founded StratifAI to translate these novel ideas into patient care.

Our products enable cancer patients to receive the right treatment at the right time - both for existing drugs on the market, and by supporting pharmaceutical companies in developing the drugs of tomorrow.