Senior Machine Learning Engineer - Generative UI Platform

Requirements

• You have strong experience in machine learning, with a focus on applied AI systems in production environments
• You have worked on problems involving transforming text, design artifacts, or multimodal inputs into structured outputs
• You are experienced with agent-based systems and LLM orchestration frameworks (e.g., Claude Agent SDK or similar)
• You have experience working with design systems, UI data, or adjacent domains (e.g., Figma APIs, front-end structures, or visual representations)
• You are experienced with modern ML stacks and can work across Typescript and/or Python

What You'll Be Doing

• Shape the machine learning direction for UI Studio as it evolves into a generative UI platform
• Build systems that translate natural language into fully functional UI experiences through applied agent chat
• Design and implement agent-based architectures as the core interaction layer for UI generation
• Lead work on computational understanding of UI — transforming design files, product specs, and prompts into structured, machine-readable representations
• Develop pipelines that convert unstructured inputs into structured UI blueprints

Perks and Benefits

• This role is based in Stockholm
• We offer you the flexibility to work where you work best! There will be some in-person meetings, but it still allows for flexibility to work from home.