Machine Learning Engineer (m/w/d) - Foundation Models
Core
Develop and train domain-specific vision foundation models for waste sorting applications using state-of-the-art architectures and proprietary image datasets.
Role type
Senior Machine Learning Engineer (Computer Vision & Foundation Models)
Builds
Pretrained vision models, evaluation infrastructure, and optimized inference pipelines for edge hardware.
Domain
Computer Vision, Foundation Models, Waste Management/Recycling
Deliverable
production ML models
Required skills
PyTorch (distributed training, mixed precision), Computer Vision (Transformers, Self-Supervised Learning), Model Evaluation (Linear Probing, Few-Shot Detection), Model Distillation, Edge Deployment (ONNX, TensorRT), Cloud GPU Infrastructure, Data Pipeline Optimization
Preferred skills
Research experience (papers, open-source), PhD, Anomaly Detection frameworks, Experiment Tracking tools
Technologies
PyTorch 2.x, FSDP, DDP, Hydra, W&B, DVC, timm, ONNX, TensorRT, OpenVINO, AWS/Azure/GCP
Responsibilities
Design and train domain-specific vision models, build and maintain evaluation suites, optimize models for edge hardware deployment, analyze training runs to resolve issues like feature collapse, collaborate on model deployment to cloud backend, translate research papers into production solutions.
Seniority
Senior, hands-on IC