CareerPlanSign in

Senior Machine Learning Engineer

Sweden💼 Full-time🗓 2026-06-29 → 2026-09-26

Core

Optimizing training and deployment of ML models for mobile devices to enhance data privacy and customer experience, with a focus on Federated Learning.

Role type

Senior Machine Learning Engineer (Federated Learning & Mobile)

Builds

Production-ready Federated Learning systems and on-device ML pipelines for mobile applications

Domain

Mobile technology, Privacy-preserving machine learning, NLP

Deliverable

production ML models

Required skills

Federated Learning frameworks (TensorFlow Federated, Flower, FedML, OpenFL), distributed machine learning, NLP and language models, on-device ML deployment (TensorFlow Lite, Core ML, ONNX Runtime Mobile), TensorFlow, PyTorch, ML experiment design, privacy-preserving ML concepts

Preferred skills

Production FL system operation, MLOps and automated pipelines, fraud detection or trust & safety domain experience

Technologies

TensorFlow Federated, Flower, FedML, OpenFL, TensorFlow Lite, Core ML, ONNX Runtime Mobile, TensorFlow, PyTorch

Responsibilities

Lead technical evaluation and implementation of Federated Learning initiatives, design and validate end-to-end FL workflows with Data Science and Android teams, develop and optimize language models and on-device training pipelines, establish model evaluation frameworks and validation strategies, identify technical risks and provide architecture recommendations, shape the roadmap for scaling FL from experimentation to production

Seniority

Senior, hands-on IC

Sourced via jobtech · Listed on CareerPlan, which tracks 70,000+ jobs from 20+ sources.