Senior ML Engineer - Federated Learning on Mobile Stockholm

Hybrus AB; Sweden; Hybrid; SEK 600,000 - 800,000; Full time

Job summary

Hybrus AB is seeking a Senior ML Engineer for a long-term assignment in Stockholm. In this role, you will lead the implementation of Federated Learning projects, optimizing ML models for mobile platforms to enhance privacy and customer experience.

The ideal candidate has 5+ years of experience in ML Engineering, expertise in Federated Learning frameworks, and a strong background in deploying models on mobile devices. This position offers potential hybrid work options.

Qualifications

• 5+ years of experience in Machine Learning Engineering or related fields.
• Hands-on experience with Federated Learning frameworks.
• Strong understanding of model training and evaluation techniques.
• Experience deploying ML models on mobile devices.
• Ability to communicate complex technical concepts effectively.

Responsibilities

• Lead technical evaluation and implementation of Federated Learning initiatives.
• Work with teams to design end-to-end FL workflows.
• Define and execute experimentation plans for FL.
• Develop language models and on-device training pipelines.
• Shape roadmap for scaling FL from experimentation to production.

Skills

Machine Learning Engineering

Federated Learning frameworks

TensorFlow

Mobile deployment

NLP

Tools

TensorFlow Lite

PyTorch

FedML

OpenFL

Job description

Hybrus AB is seeking a Senior ML Engineer for a long-term assignment in Stockholm. In this role, you will lead the implementation of Federated Learning projects, optimizing ML models for mobile platforms to enhance privacy and customer experience.

The ideal candidate has 5+ years of experience in ML Engineering, expertise in Federated Learning frameworks, and a strong background in deploying models on mobile devices. This position offers potential hybrid work options.