Medical ML/AI Engineer

Compremium AG; Bern; Hybrid; CHF 120,000 - 190,000; Full time

Job summary

Compremium AG in Bern, Switzerland seeks a motivated Medical ML/AI Engineer to develop and deploy AI-driven components for medical products. The role spans data prep, model development, validation, optimization, on-device deployment and ongoing improvement in a regulated environment.

The candidate will work across data quality, regulatory compliance, and cross-functional teams to ensure robust, explainable AI aligned with ISO 13485/IEC 62304 and FDA guidance.

Qualifications

• Master’s degree or PhD in ML/AI or related field.
• 2+ years of ML/AI experience in regulated medical environment.
• Strong Python and ML framework experience (TensorFlow, PyTorch).
• C/C++ knowledge for integration or inference pipelines.
• Experience with multimodal medical data, including imaging (ultrasound).

Responsibilities

• Designing, developing, training, and validating ML/AI models for medical products.
• Collaborating with data teams to ensure data quality and regulatory compliance.
• Driving system-level AI architecture and model feasibility, performance, scalability.
• Optimizing models for on-device/embedded deployment (compression, quantization).
• Testing, evaluation, robustness analysis, and validation on clinical datasets.
• Documenting architectural, validation and regulatory activities per ISO 13485, IEC 62304, FDA guidance.
• Collaborating with software, clinical, and regulatory teams to align AI with product needs.

Skills

ML/AI development

Python

TensorFlow

PyTorch

C/C++

Regulatory compliance

English fluency

Education

Master’s degree or PhD in ML/AI/CS/Biomedical Eng

Tools

ONNX

Job description

Compremium is seeking a motivated Medical ML / AI Engineer to join our multidisciplinary team and contribute to the development and deployment of AI-driven components for our medical products. The ideal candidate has a strong background in artificial intelligence and firsthand experience working with multimodal medical data (including imaging) in a regulated environment. The successful candidate will play a key role across the full AI product lifecycle, from data preparation and model development to validation, optimization, on-device deployment and continuous improvement, ensuring that AI solutions are robust, explainable, and compliant with applicable medical device regulations.

Join us in our journey and make a meaningful impact on the lives of patients and healthcare professionals!

• Designing, developing, training, and validating machine learning and AI models for medical products
• Collaborating closely with the data team to ensure data quality, traceability, proper labeling, and compliance with regulatory requirements
• Driving system-level AI architecture and discussing model feasibility, performance, scalability, and maintainability
• Optimizing AI models for deployment on embedded or edge devices, including model compression, quantization, and acceleration techniques
• Testing and verification of AI components, including performance evaluation, robustness analysis, and validation on clinical or pre-clinical datasets
• Contributing to technical, algorithmic, validation and continuous improvement documentation in line with ISO 13485, IEC 62304, FDA guidance documents, the EU AI Act and other emerging AI-related and medical regulatory standards
• Collaborating cross-functionally with software, clinical, and regulatory teams to define data collection strategies and align AI solutions with clinical needs and product requirements
• Master’s degree or PhD in Machine Learning, Artificial Intelligence, Computer Science, Biomedical Engineering, or a related field
• 2+ years of professional experience in ML/AI development, including experience in a regulated medical environment
• Strong proficiency in Python and common ML/AI frameworks (TensorFlow, PyTorch, or similar)
• Working knowledge of C/C++ for model integration or inference pipelines
• Experience with model export and deployment workflows (e.g. ONNX)
• Solid knowledge of model optimization for on-device or embedded execution
• Previous experience working with multimodal medical data, including imaging (especially ultrasound) data
• Fast-learning and proactive mindset
• Excellent written and spoken English; German working proficiency is a plus
• On-site (in Muri bei Bern) with possibility of partial remote work

Required Knowledge, Skills, & Abilities

• Deep understanding of modern software development practices, including modular design, automated testing, code review, and maintainable production code
• Work experience with agentic coding workflows and AI-assisted software-development tools, with appropriate human review, validation, traceability, security, and regulatory oversight
• Experience with CI/CD, MLOps, and version control for software, data and models
• Ability to work independently and as part of a team, take ownership of technical deliverables, and collaborate effectively within a multidisciplinary team
• Demonstrates passion and enthusiasm for the company’s vision and work
• Ability to adapt to a changing, fast-paced environment

About Compremium AG

Compremium is a medical diagnostics company based in Bern, Switzerland. We are on a mission to transform the diagnosis of pressure-related conditions in the human body with our first-of-its-kind, non-invasive platform technology. Our first product is cleared by the FDA and CE and is currently used by multiple reputed institutions in the United States and Europe. Our second product has recently been granted FDA breakthrough designation and is being used by NASA in the International Space Station. We are launching our products in the United States and Europe, with a rapid expansion plan to scale quickly.