Machine Learning Engineer - Zurich, Switzerland

We are looking for a computer vision and machine learning (CV/ML) engineer to enhance the perception capabilities of our fully autonomous drone-in-a-box system. You will be joining a small, CV/ML focused team and be one of the core contributors to our software stack, taking algorithms and models from prototype to deployment on the systems in the field. This position is for an on-site employee for our Zurich office.

• Design and develop a wide variety of computer vision and machine learning features for real-time unmanned aerial vehicle (UAV) applications
• Train, fine-tune and evaluate deep learning models for tasks such as object detection, semantic segmentation, image retrieval/reidentification
• Optimize algorithms and models for edge devices (NVIDIA Jetson and Qualcomm Dragonwing), and integrate them into our image and video processing pipelines
• Monitor ML model performances in real world deployments and set up workflows for continuous improvement (data collection, curation, labeling, retraining, evaluation)
• Integrate large/vision language models into our system, and design, test and maintain queries for tasks like scene understanding, situational awareness and hazard analysis

• Degree in Computer Science, Electrical Engineering, Robotics Engineering, or a related field (or equivalent practical experience)
• Solid understanding of computer vision and machine learning fundamentals
• Strong programming skills in Python and C++
• Experience with PyTorch or TensorFlow, OpenCV, ROS/ROS2, CUDA, TensorRT
• Experience with end-to-end model development: from dataset curation, labeling strategies, active learning workflows to deployment and optimization for edge devices
• Experience with transformers, large/vision language models and prompt design for perception tasks is a plus
• Experience with GStreamer pipelines and custom plugin development is a plus
• Experience with Qualcomm edge devices and CV/ML SDKs is a plus
• Good oral and written skills in English
• Work permit in Switzerland required