Machine Learning Engineer (f/m/d)

Wir suchen einen ehrgeizigen und analytischen Machine Learning Engineer (f/m/d) mit Sitz in Berlin. In dieser Rolle werden Sie unsere KI-Initiativen vorantreiben, indem Sie fortschrittliche Computer Vision-Modelle für die medizinische Bildanalyse entwerfen, implementieren und warten.

Diese Rolle ist ideal für Sie, wenn Sie einen starken technischen Hintergrund im Bereich Machine Learning mit praktischer Softwareentwicklungserfahrung kombinieren und bereit sind, in einem schnell wachsenden Health-Tech-Startup echten Einfluss zu nehmen - in enger Zusammenarbeit mit unseren interdisziplinären AI-, Produkt- und Geschäftsteams.

• Entwerfen, Trainieren und Evaluieren von Computer Vision-Modellen für die medizinische Bildanalyse mit PyTorch und PyTorch Lightning.
• Überwachen von Produktionsmodellen, Optimieren von Leistungskennzahlen und proaktives Implementieren von Retraining-Strategien zur Bekämpfung von Modellveränderungen.
• Datenvorverarbeitung, -erweiterung und Qualitätskontrolle in enger Zusammenarbeit mit Datenannotatoren und medizinischen Beratern durchführen.
• Systematische Pflege organisierter Datensätze, Modellversionierung (FiftyOne, DVC), Experimentverfolgung (MLflow) und automatisierte Tests.
• Aufbauen und Warten skalierbarer ML-Pipelines und APIs (FastAPI/Flask) unter Verwendung von CI/CD-Pipelines, Containerisierung (Docker) und MLOps-Praktiken für automatisierte Workflows.
• Eng mit Frontend- und Backend-Entwicklern zusammenarbeiten, um KI-Modelle reibungslos in Produkt-Workflows zu integrieren.
• ELT-Pipelines (Meltano) aufbauen und warten, Workflows orchestrieren (Airflow) und Dashboards sowie Visualisierungen (Metabase) für umsetzbare Modellinsights und Geschäftszahlen erstellen.
• Sicherstellen einer hohen Codequalität durch Versionskontrolle (Git), gründliche Code-Reviews und umfassende Dokumentation von Modellen, Experimenten und Best Practices.
• Interdisziplinär innerhalb der AI-, Produkt- und Geschäftsteams arbeiten und Ihre Ergebnisse sowohl technischen als auch nicht-technischen Stakeholdern effektiv präsentieren.

• Sie haben einen Bachelor- oder Masterabschluss in Informatik, Machine Learning, Mathematik oder einem verwandten Bereich.
• Sie bringen mehr als 2 Jahre praktische Berufserfahrung im Bereich Machine Learning mit und haben eine nachweisliche Erfolgsbilanz mit modernen ML-Frameworks (PyTorch, TensorFlow).
• Idealerweise haben Sie praktische Erfahrung in Computer Vision, NLP (Natural Language Processing), Zeitreihenanalyse oder Reinforcement Learning.
• Sie verfügen über starke Programmierkenntnisse in Python und sind sehr vertraut mit wissenschaftlichen Rechenbibliotheken (NumPy, Pandas, Scikit-learn).
• Sie haben solide Erfahrungen mit Versionskontrolle (Git), kollaborativer Entwicklung und Python-Paketierung sowie Abhängigkeitsmanagement (Poetry, uv).
• Sie sind erfahren in der Docker-Containerisierung, wenden MLOps-Praktiken (MLflow, DVC) an und sind versiert im Umgang mit Datenengineering-Tools (Meltano, Airflow, Metabase).
• Sie bringen Webentwicklungskompetenzen (FastAPI, Flask, Django) mit, um robuste ML-APIs zu erstellen, und haben Erfahrung in der Entwicklung von CI/CD-Pipelines für ML-Workflows.
• Sie sprechen fließend Englisch und besitzen ausgezeichnete Kommunikationsfähigkeiten, um in einem interdisziplinären Umfeld erfolgreich zu sein.

• Wir investieren in Sie. Wir bieten eine steile Lernkurve und die Möglichkeit, durch kontinuierliches Feedback und eine Kultur des lebenslangen Lernens zu wachsen.
• Wir geben Ihnen die Freiheit, unabhängig zu entwickeln. Ihre Energie und Expertise werden darauf gerichtet, das Leben von Menschen mit chronischen Erkrankungen zu verbessern.
• Ein monatliches Budget, das Sie frei über unsere Benefits-App nutzen können – für Fitnessstudio-Mitgliedschaften, Wellness oder andere Aktivitäten.
• Attraktives Büro in bester Lage mit Blick auf die Spree im Herzen von Kreuzberg, Berlin.

Nia Health GmbH, unterstützt von institutionellen VCs, entwickelt und vermarktet KI-gestützte medizinische Software, um umfassende digitale Unterstützung für Menschen mit chronischen Erkrankungen bereitzustellen. Das Health-Tech-Unternehmen Nia Health wurde als Spin-off der Charité – Universitätsmedizin Berlin gegründet. Das erste Produkt, die preisgekrönte medizinische Atopische Dermatitis-App Nia, bietet täglich Unterstützung für Tausende von Patienten und deren Familien und ist die am häufigsten verwendete App für atopische Dermatitis im deutschsprachigen Raum. Nia Health arbeitet mit führenden Pharmaunternehmen, Krankenhäusern und Krankenkassen zusammen.