Senior ML Engineer — Federated Learning & On-Device AI

Hybrus AB; Stockholms kommun; Hybrid; SEK 837,000 - 1,116,000; Full time

Job summary

Hybrus AB is seeking a Senior ML Engineer for a long-term project in Stockholm. In this role, you will optimise training and deployment of ML models, focusing on mobile integration to enhance data privacy. You will collaborate with various teams to implement best practices in ML model development and deployment, particularly in Federated Learning.

The position requires strong hands-on experience and a deep understanding of machine learning frameworks. This role offers a dynamic work environment aimed at making significant impacts in the field.

Qualifications

• 5+ years of experience in Machine Learning Engineering or related fields.
• Hands-on experience with Federated Learning frameworks.
• Strong problem-solving skills in exploratory research environments.

Responsibilities

• Lead the technical evaluation of Federated Learning initiatives.
• Collaborate with Data Science and Mobile teams to design workflows.
• Develop on-device training pipelines for privacy-preserving ML.

Skills

Machine Learning Engineering

Federated Learning frameworks

Distributed machine learning

On-device machine learning

NLP and language models

Tools

TensorFlow

PyTorch

Job description

Hybrus AB is seeking a Senior ML Engineer for a long-term project in Stockholm. In this role, you will optimise training and deployment of ML models, focusing on mobile integration to enhance data privacy. You will collaborate with various teams to implement best practices in ML model development and deployment, particularly in Federated Learning.

The position requires strong hands-on experience and a deep understanding of machine learning frameworks. This role offers a dynamic work environment aimed at making significant impacts in the field.