[Hiring] ML Engineer @StellDirVor GmbH

Role Description

As our ML Engineer (Computer Vision & ML Ops), you will build the AI capabilities at the core of ARAIAS:

• Evaluating and fine-tuning computer vision models on smart glasses data.
• Running the ML operations behind them.
• Applying generative AI to create synthetic medical images for education.
• Owning the full data lifecycle end to end — from sourcing, consent and annotation to preparation — alongside model development and deployment.

We're at an early stage — moving from concept to working prototype — so the near-term focus is on validating models and building toward a proof of concept we can test in real education and clinical settings.

Tasks

• Select and benchmark pre-trained models and/or validated architectures for medical image analysis and adapt them to domain-specific data, ensuring robust performance under real-world conditions.
• Identify and adapt suitable generative AI models to create realistic synthetic medical images for education and training.
• Evaluate and fine-tune small language models (SLMs) for voice interaction — speech-to-text, intent understanding and text-to-speech — optimized for low-latency or on-device use with smart glasses.
• Research and assess state-of-the-art ML models and emerging trends, including Explainable AI (XAI), cybersecurity for ML, and deployment and distribution modes (edge vs. cloud, federated learning, model compression).
• Plan and execute the ML operations pipeline — data gathering, annotation, preparation, training, testing and deployment.
• Propose a data collection strategy — research suitable public and clinical data sources, recommend patient consent and other administrative requirements for ML training (e.g., data protection agreements, ethics approval), and define collection and annotation protocols with clinical experts, including labeling guidelines and quality checks.
• Identify and validate factors that affect model performance — image conditions, dataset bias, patient diversity, data privacy and regulatory requirements.
• Define test strategies and metrics, and validate model outputs against clinician assessments.
• Perform build vs. buy analyses for models, tools and services, including cost comparison.
• Estimate compute and training costs and propose budgets for model training and deployment.
• Choose and set up training environments flexibly — local hardware or cloud — based on cost, data privacy and performance.
• Make pragmatic technical decisions appropriate for the current prototyping stage and document results clearly.
• Maintain clear, traceable documentation of data sources, models, experiments and design decisions to support audits and quality reviews.
• Evaluate and integrate additional sensors where feasible (e.g., depth, LiDAR, thermal, multispectral) and fuse them with RGB input.

Qualifications

• Degree in Computer Science, Data Science, Biomedical Engineering, Mathematics or a related field — or equivalent practical background.
• Strong Python skills and hands-on use of PyTorch, OpenCV or TensorFlow.
• Solid understanding of computer vision — classification, object detection and image segmentation, 3D Reconstruction and Multi-View Geometry.
• Hands-on knowledge of transfer learning — adapting pre-trained models and fine-tuning them on custom datasets.
• Working knowledge of the full ML lifecycle — data preparation, training, evaluation and deployment.
• Basic knowledge of generative models (diffusion models or GANs) and common open-source libraries.
• Comfortable training models both locally (GPU workstation) and on cloud platforms.
• Able to estimate ML compute costs and compare build vs. buy options to support budget decisions.
• Aware of medical imaging challenges — limited data, label variability between clinicians and class imbalance.
• Aware of healthcare data privacy and ethics (e.g., GDPR, patient consent, anonymization) and why explainability and bias matter in clinical AI.
• Able to read research papers and apply them pragmatically.
• Self-directed and pragmatic — able to take initiative and own problems end-to-end in a lean team.
• Fluent in English — German is a bonus.

Requirements

• Projects or coursework in medical imaging or clinical AI.
• Familiarity with semi-supervised learning — leveraging limited or partially labeled data.
• Experience with 3D data processing — depth maps, point clouds or mesh reconstruction.
• Exposure to multi-modal learning — combining RGB and depth or LiDAR data.
• Edge AI or wearable deployment (e.g., TensorFlow Lite, ONNX, model quantization).
• Knowledge of explainability methods (e.g., Grad-CAM, SHAP) or adversarial robustness.
• Familiarity with MLOps tools (e.g., MLflow, Weights & Biases, DVC, Docker).
• Aware of medical device regulation for AI software (e.g., EU MDR, IEC 62304).
• Familiarity with small language models (SLMs) and speech pipelines (e.g., Whisper, on-device speech recognition and text-to-speech).

Benefits

• Meaningful work — You help shape how technology redefines education and training in healthcare, with real impact on clinical outcomes.
• Ownership and creative freedom — You work autonomously, contribute your ideas directly, and have a genuine say in how the system is built.
• A learning culture — We support personal and professional growth through knowledge sharing, access to innovative tools and methods, and a dedicated learning budget.
• Flexible working — Remote-friendly setup with the option to work from our Munich office.
• Flat hierarchies and team spirit — A small, open, and trust-based team where communication is direct and everyone's voice matters.
• International collaboration — Work alongside a Taiwan-based partner covering technology, hardware and research.