Computer Vision and Machine Learning Engineer

Sunflower Labs; Stallikon; On-site; CHF 120,000 - 170,000; Full time

Job summary

Sunflower Labs is seeking a CV/ML engineer to enhance perception in our autonomous drone-in-a-box system in Zurich. This on-site role joins a small, CV/ML‑focused team responsible for turning prototypes into deployable software.

You will design and optimize real‑time vision and learning components, train models, and integrate edge‑device pipelines (Jetson/Dragonwing). English communication and Swiss work permit are required.

Qualifications

• 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: dataset curation, labeling strategies, active learning workflows to deployment and optimization for edge devices.

Responsibilities

• Design and develop a wide variety of computer vision and machine learning features for real-time 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.

Skills

Python

C++

OpenCV

PyTorch/TensorFlow

ROS/ROS2

CUDA/TensorRT

Edge devices

English

Education

Bachelor's degree in CS/EE/Robotics

Tools

GStreamer

Jetson platform

TensorRT

CUDA

Qualcomm edge devices

Job description

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.

Tasks

• 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

Requirements

• 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