CareerPlanSign in

Machine Learning Engineer

Stockholm, Sweden💼 Full-time🗓 2026-07-05 → 2026-09-26

Core

Building the core models for a data infrastructure platform that determines AI prediction reliability and routes cases for automation or human review.

Role type

Senior IC machine learning engineer (deep learning & GNNs)

Builds

Production-grade deep learning models and platform infrastructure

Domain

AI reliability, data infrastructure, deep learning

Deliverable

production ML models

Required skills

deep learning fundamentals, model training and evaluation, PyTorch, mathematical reasoning, model architecture design, statistical foundations (Conformal Prediction, calibration), code quality

Preferred skills

Graph Neural Networks, PyTorch Geometric/DGL, open-source contributions, research-to-production experience

Technologies

PyTorch, PyTorch Geometric, DGL

Responsibilities

Train, evaluate, and improve platform models; diagnose model behavior using underlying mathematics; make principled model design choices; collaborate with engineering to deliver research-grade models to production

Seniority

Mid-level, hands-on IC

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