ML Platform Engineer (m/f/d)

Die AI Research Division von Agile Robots sucht einen ML Platform Engineer (m/w/d), der die verteilte Trainings-, Bereitstellungs- und Experimentierinfrastruktur aufbaut und betreibt, auf die Forschungs-, Daten- und Robotikteams angewiesen sind, um Modelle von Prototypen in die Produktion zu überführen.

Ihre Verantwortlichkeiten

• Training Infrastructure: Entwerfen und Skalieren verteilter Trainingsworkflows für große Modelle mit Tools wie PyTorch Distributed, DeepSpeed und Cluster-Schedulern wie SLURM oder Kubernetes.
• ML Platform: Aufbau und Wartung containerisierter ML-Umgebungen, die reproduzierbare Experimente und Benchmarking unterstützen.
• CI/CD Pipelines: Entwicklung und Wartung von CI/CD-Pipelines für maschinelles Lernen, um zuverlässiges Testen, Training und Bereitstellung von Modellen zu ermöglichen.
• Lifecycle Management: Implementierung von Experimentverfolgung, Modellversionierung und Reproduzierbarkeits-Workflows mit Tools wie ClearML oder Weights & Biases.
• Observability: Einrichtung von Überwachungssystemen wie Prometheus und Grafana zur Verfolgung der Modellleistung und Systemgesundheit sowie zur Erkennung von Drift in der Produktion.
• Cross-Team Collaboration: Zusammenarbeit mit Forschungs-, Daten- und Robotikteams, um neue Modelle mit robusten Produktionssystemen zu verbinden.

Wesentliche Fähigkeiten

• Hintergrund und Erfahrung: Abschluss in Informatik, Software Engineering oder einem verwandten Bereich, mit beruflicher Erfahrung im Aufbau und Betrieb von ML- oder Softwareinfrastrukturen in der Produktion.
• Distributed Training: Erfahrung im Entwerfen und Betreiben verteilter Trainingssysteme auf Kubernetes und Docker, unter Verwendung von PyTorch Distributed, DeepSpeed und Planern wie SLURM.
• CI/CD für ML: Erfahrung im Aufbau von CI/CD-Pipelines, die zuverlässiges Testen, Training und Bereitstellung von Modellen unterstützen.
• Cloud Infrastructure: Erfahrung im Betrieb von ML-Workloads auf Cloud-Infrastrukturen, vorzugsweise AWS.
• Experiment Tracking: Praktische Erfahrung mit Experimentverfolgung und Modellversionierung unter Verwendung von Tools wie MLflow oder Weights & Biases.
• Observability: Erfahrung mit Überwachung und Drift-Erkennung unter Verwendung von Tools wie Prometheus und Grafana.
• Software Engineering: Python- und Systemdesignfähigkeiten, mit Erfahrung im Aufbau und Betrieb von ML-Systemen über die Prototypenphase hinaus.

Wünschenswerte Fähigkeiten

• Multimodal Systems: Erfahrung mit großangelegten oder multimodalen ML-Systemen wie Vision-Language-Action-Modellen.
• Infrastructure As Code: Vertrautheit mit Infrastructure-as-Code-Tools wie Terraform.
• ML Orchestration: Erfahrung mit ML-Pipeline- und Orchestrierungstools.
• Distributed Compute: Kenntnisse in Hochleistungs- oder verteilten Rechenumgebungen.

Was wir bieten

• Dynamisches High-Tech-Unternehmen kombiniert mit finanzieller Solidität und erstklassigen Investoren.
• Ein interdisziplinäres, internationales Team mit über 60 verschiedenen Nationalitäten in einer kollaborativen Arbeitsumgebung.
• Viele Entwicklungsmöglichkeiten im Kontext unseres kontinuierlichen Wachstums.
• Herausfordernde Aufgaben und wirkungsvolle Projekte neben Experten, die professionelles und persönliches Wachstum ermöglichen.
• Corporate Benefits Programm, das Gesundheit, Mobilität und Lernen mit 100 € netto pro Monat abdeckt.
• Moderne Büros mit Dachterrasse mit Blick auf München, kostenlose Getränke und Obst sowie regelmäßige Unternehmensveranstaltungen tragen zu einem guten Arbeitsumfeld bei.

Über uns

Agile Robots SE ist ein internationales High-Tech-Unternehmen mit Sitz in München, Deutschland, mit einem Produktionsstandort in Kaufbeuren und mehr als 2300 Mitarbeitern weltweit. Unsere Mission ist es, die Lücke zwischen künstlicher Intelligenz und Robotik zu schließen, indem wir Systeme entwickeln, die modernste Kraft-Moment-Sensorik und führende Bildverarbeitungstechnologie kombinieren. Diese einzigartige Kombination von Technologien ermöglicht es uns, benutzerfreundliche und erschwingliche Roboterlösungen anzubieten, die eine intelligente Präzisionsmontage ermöglichen. Dies wird durch unsere Mitarbeiter möglich, die jeden Tag ihr Bestes mit Kreativität und Begeisterung geben. Werden Sie Teil dieses Teams und gestalten Sie die Zukunft der Robotik mit uns!

Wir sind stolz auf unsere Vielfalt und freuen uns auf Ihre Bewerbung, unabhängig von Geschlecht und sexueller Identität, Nationalität, Ethnie, Religion, Alter oder Behinderung.