Machine Learning Engineer (f/m/d)

Beschreibung

Wir suchen einen ambitionierten und analytisch denkenden Machine Learning Engineer (f/w/d) an unserem Standort in Berlin. In dieser Rolle treibst du unsere KI-Initiativen voran, indem du fortschrittliche Computer-Vision-Modelle für die medizinische Bildanalyse entwirfst, bereitstellst und wartest.
Diese Rolle ist ideal für dich, wenn du einen starken technischen Hintergrund im Bereich Machine Learning mit praktischer Erfahrung im Software Engineering kombinierst. Wenn du Lust hast, in einem schnell wachsenden Health-Tech-Startup echten Impact zu schaffen und eng mit unseren funktionsübergreifenden KI-, Produkt- und Business-Teams zusammenzuarbeiten, bist du bei uns genau richtig.

Deine Aufgaben

• Entwirf, trainiere und evaluiere Computer-Vision-Modelle für die medizinische Bildanalyse mithilfe von PyTorch und PyTorch Lightning.
• Überwache Produktionsmodelle, optimiere Leistungskennzahlen und implementiere proaktiv Retraining-Strategien, um Model Drift entgegenzuwirken.
• Führe Datenvorverarbeitung, Data Augmentation und Qualitätskontrollen in enger Zusammenarbeit mit Data Annotators und medizinischen Beratern durch.
• Pflege systematisch organisierte Datensätze, Modellversionierung (FiftyOne, DVC), Experiment-Tracking (MLflow) und automatisierte Tests.
• Entwickle und pflege skalierbare ML-Pipelines und APIs (FastAPI/Flask) unter Einsatz von CI/CD-Pipelines, Containerisierung (Docker) und MLOps-Praktiken für automatisierte Workflows.
• Arbeite eng mit Frontend- und Backend-Entwicklern zusammen, um KI-Modelle reibungslos in Produkt-Workflows zu integrieren.
• Erstelle und pflege ELT-Pipelines (Meltano), orchestriere Workflows (Airflow) und entwickle Dashboards sowie Visualisierungen (Metabase) für handlungsorientierte Modelleinblicke und Geschäftskennzahlen.
• Stelle eine hohe Codequalität durch Versionskontrolle (Git), gründliche Code-Reviews und umfassende Dokumentation von Modellen, Experimenten und Best Practices sicher.
• Arbeite funktionsübergreifend innerhalb der KI-, Produkt- und Business-Teams und präsentiere deine Ergebnisse überzeugend vor technischen und nicht-technischen Stakeholdern.

Dein Profil

• Du hast einen Bachelor- oder Masterabschluss in Informatik, Machine Learning, Mathematik oder einem verwandten Bereich.
• Du bringst mehr als 2 Jahre praktische Berufserfahrung im Bereich Machine Learning mit und hast nachweisliche Erfolge in der Arbeit mit modernen ML-Frameworks (PyTorch, TensorFlow).
• Du hast idealerweise praktische Erfahrung im Bereich Computer Vision, NLP (Natural Language Processing), Zeitreihenanalyse oder Reinforcement Learning.
• Du verfügst über sehr gute Programmierkenntnisse in Python und bist bestens vertraut mit wissenschaftlichen Programmbibliotheken (NumPy, Pandas, Scikit-learn).
• Du verfügst über fundierte Erfahrung mit Versionskontrolle (Git), kollaborativer Entwicklung sowie Python-Packaging und Dependency-Management (Poetry, uv).
• Du bist erfahren in der Docker-Containerisierung, wendest MLOps-Praktiken (MLflow, DVC) sicher an und bist versiert im Umgang mit Data-Engineering-Tools (Meltano, Airflow, Metabase).
• Du hast Webentwicklungs-Skills (FastAPI, Flask, Django) für den Aufbau robuster ML-APIs und besitzt Erfahrung in der Entwicklung von CI/CD-Pipelines für ML-Workflows.
• Du sprichst fließend Englisch (gute Deutschkenntnisse sind ein Plus) und verfügst über ausgezeichnete Kommunikationsfähigkeiten, um in einem funktionsübergreifenden Umfeld erfolgreich zu sein.

Warum wir? Deine Benefits

• Wir investieren in dich. Wir bieten dir eine steile Lernkurve und die Möglichkeit, durch kontinuierliches Feedback und eine Kultur des lebenslangen Lernens zu wachsen. Durch unsere Partnerschaft mit adesso hast du Zugang zu einer Auswahl von über 100 Schulungen.
• Gleichzeitig geben wir dir den Freiraum, dich eigenständig zu entwickeln. Deine Energie und Expertise fließen direkt in die Verbesserung der Lebensqualität von Menschen mit chronischen Erkrankungen. Bei uns arbeitest du an etwas, das wirklich einen Unterschied macht!
• Ein monatliches Budget, das du über unsere Benefits-App frei nutzen kannst – ob für die Fitnessstudio-Mitgliedschaft, Wellness oder andere Aktivitäten.
• Ein attraktives Büro in bester Lage mit Blick auf die Spree im Herzen von Berlin-Kreuzberg.