Machine Learning Engineer (m/w/d) - Foundation Models

WeSort.AI GmbH · Würzburg, Bavaria, Germany

Location
Würzburg
Job Type
CDI
Posted
July 08, 2026

Job Description

Das ist deine neue Leidenschaft



  • Du entwickelst und trainierst unsere domänenspezifischen Vision-Modelle auf Basis aktueller State-of-the-Art-Architekturen und nutzt unsere Abfallbild-Datenbasis für Pretraining und Continued Pretraining

  • Du gestaltest unsere komplette ML-Trainings-Pipeline: von der Datenaufbereitung über verteiltes Training (PyTorch FSDP/DDP, Mixed Precision) bis zur Modell-Versionierung

  • Du baust und pflegst unsere Eval Suite – die zentrale Infrastruktur, die misst, ob unsere Modelle wirklich besser werden: Linear Probing, k-NN-Probing, Few-Shot-Detection, Cross-Domain-Generalization, Anomalie-Detection

  • Du finetunst und destillierst unsere Modelle für konkrete Downstream-Tasks und Edge-Hardware (Sortieranlagen, GPU-Inferenz)

  • Du analysierst Trainings-Runs systematisch, identifizierst Probleme wie Feature Collapse oder Domain Shift und entwickelst nachhaltige Lösungen statt kurzfristiger Workarounds

Ready to Apply?

Submit your application for Machine Learning Engineer (m/w/d) - Foundation Models at WeSort.AI GmbH

Apply Now