Machine Learning Engineer (m/f/d) VLA Training & Evaluation in München

Die KI-Teams bei Agile Robots suchen einen Machine Learning Engineer (m/w/d) VLA Training & Evaluation, der die Frameworks für das Fine-Tuning und Benchmarking von Robotermodellen auf gesammelten Roboterdaten entwickelt. Die Frameworks sollen über verschiedene Grundmodelle hinweg funktionieren und für Teams mit bescheidenen Daten und Rechenressourcen geeignet sein.

Verantwortlichkeiten

• Training Workflows: Aufbau wiederverwendbarer, modellunabhängiger Trainings- und Evaluierungs-Workflows mit Experimentverfolgung, Modellversionierung und Checkpoint-Management, sodass verschiedene Grundmodelle ohne Neubau der Pipeline ausgetauscht werden können.
• Fine-Tuning: Implementierung des Fine-Tunings von vortrainierten Vision-Language-Action-Modellen auf gesammelten Roboterdaten, beispielsweise mit parameter-effizienten Methoden wie LoRA oder Adaptern oder mit Distillation.
• Policy Training: Aufbau von Trainingsschleifen für Action-Chunking und transformerbasierte Aktionsmodelle, Diffusionsrichtlinien und VLM/VLA-Köpfe.
• Benchmarks: Erstellung von Benchmarks mit Generalisierungs-Splits, Erfolgsmetriken und Vorlagen zur Fehleranalyse.
• Transfer Studies: Zusammenarbeit mit Simulationsingenieuren an Studien, einschließlich Metriken und Ablationen, die testen, ob Verbesserungen in der Simulationsrealität die Leistung des echten Roboters verbessern.
• Research Baselines: Umsetzung veröffentlichter Methoden in funktionierende Baselines und Vorlagen, die andere erweitern können.

Wesentliche Fähigkeiten

• Akademischer Hintergrund: Abschluss in Informatik, Maschinenlernen, Robotik oder einem verwandten Bereich (Master oder PhD üblich) oder eine gleichwertige Kombination aus formaler Ausbildung und nachgewiesener Erfahrung im Training von Deep-Learning-Modellen.
• Training Internals: Praktische Erfahrung im Schreiben und Debuggen von Trainingsschleifen in PyTorch oder JAX, einschließlich Datenladen, Verlustdesign und Optimierer-Verhalten.
• Fine-Tuning: Erfahrung im Fine-Tuning vortrainierter Modelle und deren Anpassung an neue Aufgaben und Daten.
• Robot Learning: Grundlegendes Verständnis dafür, wie imitierte Lernrichtlinien, wie Action-Chunking oder Diffusionsrichtlinien, Beobachtungen in Roboteraktionen umwandeln.
• Experiment Management: Erfahrung im Entwerfen von Evaluierungen und Ablationen sowie im Durchführen von Experimenten mit Verfolgung, Versionierung und Checkpoint-Management, sodass andere diese reproduzieren können.

Wünschenswerte Fähigkeiten

• VLA-Modelle: Erfahrung im Fine-Tuning von Vision-Language-Action- oder Vision-Language-Modellen für die Robotersteuerung.
• Real-Robot-Evaluation: Erfahrung in der Evaluierung gelernter Richtlinien auf physischen Robotern.
• Sim-to-Real: Vertrautheit mit der Messung, wie gut Modelle, die in Simulationen trainiert wurden, auf echte Hardware übertragen werden.

Was wir bieten

• Ein 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, während wir weiter wachsen.
• Herausfordernde Aufgaben und wirkungsvolle Projekte neben Experten, die professionelles und persönliches Wachstum ermöglichen.
• Corporate Benefits Programm, das Gesundheit, Mobilität und Lernen für 100 € netto pro Monat abdeckt.
• Moderne Bürofazilitäten mit einer 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 3500 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 robotergestützte Lösungen anzubieten, die intelligente Präzisionsmontage ermöglichen. Dies wird durch unsere Mitarbeiter möglich gemacht, 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.