
Hybrus AB · Sweden
A Senior ML Engineer is sought to optimize ML model training and deployment for mobile devices, focusing on cost-efficiency, data privacy, and exploring federated learning.
A Senior ML Engineer is sought to optimize ML model training and deployment for mobile devices, focusing on cost-efficiency, data privacy, and exploring federated learning.
We are looking for a Senior ML Engineer for our client in Stockholm for a long term assignment.
Employment type: Fixed Term Contract
Duration : 6 Months - 1 Year
Location : Stockholm
Work type : Onsite ( potential hybrid options) As a Senior ML Engineer, you will be working hands-on to optimise training and deployment of ML models to be quick and cost-efficient. You will also be at the forefront of putting our ML models on mobile devices to enhance data privacy and customer experience. To achieve this, you will need to collaborate with teams to establish best practices and tools for efficient ML model development and deployment, particularly on mobile platforms. You will be expected to help Client reach and stay at the cutting edge of ML training and deployment, as well as explore new frontiers such as federated learning.
● Lead the technical evaluation and implementation of Federated Learning (FL) initiatives .
● Work closely with Data Science, Android, and Backend teams to design and validate end-to-end FL workflows.
● Define and execute experimentation plans to assess the effectiveness of FL for use cases.
● Develop and optimize language models and on-device training pipelines for privacy-preserving machine learning.
● Establish model evaluation frameworks, success metrics, and validation strategies for FL-based systems.
● Identify technical risks, assumptions, and limitations, and provide recommendations on architecture and future direction.
● Help shape the roadmap for scaling FL from experimentation to production-ready systems
● 5+ years of experience in Machine Learning Engineering, Applied Machine Learning, or related fields.
● Hands-on experience with Federated Learning frameworks such as TensorFlow Federated, Flower, FedML, OpenFL, or equivalent.
● Strong understanding of distributed machine learning, model training, and model evaluation techniques.
● Experience working with NLP, language models, embeddings, or text classification systems.
● Hands-on experience deploying ML models on mobile devices (e.g., TensorFlow Lite, Core ML, ONNX Runtime Mobile).
● Strong knowledge of machine learning frameworks such as TensorFlow and PyTorch.
● Experience designing and executing ML experiments, analyzing results, and driving data-driven decisions.
● Familiarity with privacy-preserving machine learning concepts and challenges.
● Ability to work across multiple teams and communicate complex technical concepts to both technical and non-technical stakeholders.
● Strong problem-solving skills and ability to operate in an exploratory research and PoC environment.
● Experience deploying or operating Federated Learning systems in production environments.
● Hands-on experience with on-device machine learning technologies such as TensorFlow Lite, ONNX Runtime Mobile, or Core ML.
● Experience building machine learning solutions for mobile applications.
● Experience in messaging, spam detection, fraud detection, trust & safety, or similar domains.
● Familiarity with the challenges of running ML workloads on mobile devices.
● Experience with MLOps, model monitoring, and automated training/deployment pipelines. Please let us know if you are interested and if yes, then apply to careers@hybrus.se with your CV, including details about Salary, Notice period and Visa Status.
💬 ACCURX IS WHERE CONVERSATIONS HAPPEN WITH AND ABOUT PATIENTS. For decades, the NHS has struggled with fragmented systems that make simple tasks feel impossible. We’re changing that by building a single, system-wide platform that connects everyone through communication. What started as a way for GPs to text a patient has now evolved into an all-in-one digital toolkit used by 98% of GP practices. Our platform now powers Total Triage to manage patient demand, and Self-Book, which lets patients schedule their own appointments in seconds. We’ve automated routine care with Patient Questionnaires for long-term conditions, while Accumail finally allows staff-to-staff communication to happen instantly across different care settings. We’re now pushing the boundaries of the consultation itself with Accurx Scribe, our AI-powered note-taker that drafts medical notes in real-time. THE TEAM We are a mission-driven team of 80 engineers based in London and the surrounding areas, united by the challenge of fixing healthcare communication. We are innovating for the NHS at a scale and depth that has never been done before, solving the real-world problems that stand between millions of patients and the care they need. We are a highly cross-functional group where engineering, product, data, and security collaborate as true peers. We foster a low-ego, high-impact environment that values expertise and new ideas, maintaining the high standards required to build and scale a national healthcare communication platform. As a Senior Machine Learning Engineer, you'll join our Triage Intelligence team — a high-impact, cross-functional group of software engineers, Data Scientists, clinicians, and a Principal Product Manager. The team builds the intelligence that powers how Accurx understands healthcare communication across the entire product suite, with a current mission to interpret patient healthcare requests and enable fast, informed decisions about the next step in a patient's care. CHALLENGES YOU’LL SOLVE... * You will take ownership of the end-to-end technical design of ML systems - from data ingestion and training pipelines through to deployment and real-time monitoring - ensuring AI features are performant, observable, and easy to debug. * You will act as the primary technical partner to our Data Scientists, evolving experimental code into high-quality, extensible library modules and building reusable components that improve developer efficiency and reduce cycle times across the team. * You will translate complex clinical and product objectives into efficient computational tasks, selecting the right tools - whether Deep Learning, NLP, or off-the-shelf - and making smart build-vs-buy trade-offs that align with Accurx's long-term technical goals. * You will help turn one-off experiments into repeatable platform capabilities, building models and systems that can be adopted by other Accurx product teams and proactively resolving cross-team dependencies as new, more efficient AI models are rolled out. YOU SHOULD APPLY IF... * You have extensive experience with a variety of ML techniques (e.g. Transformer-based NLP, Deep Learning, Tree-based methods, Bayesian modelling) and the ability to select the right tool, not just the trendiest one, for a clinical problem. * You have a proven track record of taking models from a Jupyter notebook to a high-availability production environment, managing everything from data versioning to model serving. * You have mastery of a production-grade language (e.g. Python, C#, or Go), with a focus on writing extensible, modular libraries that other teams can adopt. * You have experience defining both offline metrics (how the model performs in training) and online metrics (how the model impacts real-world user behaviour and triage outcomes). * You bring collaborative rigour to design and code reviews, surfacing erroneous assumptions before they reach production, and can preemptively identify technical risks — such as data leakage or hardware constraints — before they jeopardise the product roadmap. Bonus points if you have… * Experience working with sensitive, regulated data (GDPR, HIPAA) and implementing privacy-first techniques like differential privacy or secure data handling. * Familiarity with infrastructure-as-code tools like Terraform, Kubernetes, or MLOps platforms such as SageMaker or Vertex AI. * Hands-on experience fine-tuning LLMs, prompt engineering, or managing the latency and cost trade-offs of large language models at scale. WHAT’S IN IT FOR ME? You'll be joining an established but fast-growing Tech for Good movement, where we're led by our Principles and our mission to fix healthcare communication. * £100,000 - £115,000 salary + share options up to £30,000 * Benefits to suit you: adjust your healthcare cover, your pension or life insurance, whatever stage you’re at in life * Flexible working: We are an office first culture and ask that you’re in our (dog-friendly) Shoreditch office 3 days a week, with core hours of 10am - 4pm * Time off: You’ll get 28 days of holiday (plus bank holidays) and up to 4 weeks to work from anywhere per year * Family matters: We offer enhanced parental leave, fertility support and parental loss support * We have our very own Chef! Free healthy breakfasts, snacks and lunches will be provided, with the occasional sweet treat!
WHY FACULTY? We established Faculty in 2014 because we thought that AI would be the most important technology of our time. Since then, we’ve worked with over 350 global customers to transform their performance through human-centric AI. You can read about our real-world impact here. We don’t chase hype cycles. We innovate, build and deploy responsible AI which moves the needle - and we know a thing or two about doing it well. We bring an unparalleled depth of technical, product and delivery expertise to our clients who span government, finance, retail, energy, life sciences and defence. Our business, and reputation, is growing fast and we’re always on the lookout for individuals who share our intellectual curiosity and desire to build a positive legacy through technology. AI is an epoch-defining technology, join a company where you’ll be empowered to envision its most powerful applications, and to make them happen. ABOUT THE TEAM Our Defence team is focused on building and embedding human-centered AI solutions which give our nation a competitive edge in the defence sector. We collaborate with our clients to bring ethical, reliable and cutting-edge AI to high-stakes situations and maintain the balance of global powers essential to our liberty. Because of the nature of the work we do with our Defence clients, you will need to be eligible for UK Security Clearance (SC) and willing to work between 2 to 4 days per week on-site with these customers which may require travel to locations throughout the UK. When not required on client sites, you’ll have the flexibility to work from our London office or remotely from elsewhere within the UK. #LI-PRIO ABOUT THE ROLE As a Senior Machine Learning Engineer, we’ll look to you to lead development and deployment of cutting-edge AI systems for our diverse clients. You’ll design, build, and deploy scalable, production-grade ML software and infrastructure that meets rigorous operational and ethical standards. This is an ambitious, cross-functional role requiring a blend of technical expertise, engineering leadership, and confident client-facing skills. WHAT YOU'LL BE DOING: * Leading technical scoping and architectural decisions for high-impact ML systems * Designing and building production-grade ML software, tools, and scalable infrastructure * Defining and implementing best practices and standards for deploying machine learning at scale across the business * Collaborating with engineers, data scientists, product managers, and commercial teams to solve critical client challenges and leverage opportunities * Acting as a trusted technical advisor to customers and partners, translating complex concepts into actionable strategies * Mentoring and developing junior engineers, actively shaping our team's engineering culture and technical depth WHO WE'RE LOOKING FOR: * You understand the full ML lifecycle and have significant experience operationalising models built with frameworks like TensorFlow or PyTorch * You bring deep expertise in software engineering and strong Python skills, focusing on building robust, reusable systems * You have demonstrable hands-on experience with cloud platforms (e.g., AWS, Azure, GCP), including architecture, security, and infrastructure * You've extensive experience working with container and orchestration tools such at Docker & Kubernetes to build and manage applications at scale * You thrive in fast-paced, high-growth environments, demonstrating ownership and autonomy in driving projects to completion * You communicate exceptionally well, confidently guiding both technical teams and senior, non-technical stakeholders THE INTERVIEW PROCESS 1. Talent Team Screen (30 minutes) 2. Pair Programming Interview (90 minutes) 3. System Design Interview (90 minutes) 4. Commercial Interview (60 minutes) OUR RECRUITMENT ETHOS We aim to grow the best team - not the most similar one. We know that diversity of individuals fosters diversity of thought, and that strengthens our principle of seeking truth. And we know from experience that diverse teams deliver better work, relevant to the world in which we live. We’re united by a deep intellectual curiosity and desire to use our abilities for measurable positive impact. We strongly encourage applications from people of all backgrounds, ethnicities, genders, religions and sexual orientations. Some of our standout benefits: * Unlimited Annual Leave Policy * Private healthcare and dental * Enhanced parental leave * Family-Friendly Flexibility & Flexible working * Sanctus Coaching * Hybrid Working If you don’t feel you meet all the requirements, but are excited by the role and know you bring some key strengths, please don't hesitate in applying as you might be right for this role, or other roles. We are open to conversations about part-time hours.
Poszukujemy doświadczonego Senior Machine Learning Engineer’a