LEAD AI ENGINEER

Ants is partnering with a well-funded early-stage startup in Stockholm to find a Lead AI Engineer. The role involves designing end-to-end ML architectures, building personalization and relevance systems, and establishing ML engineering standards. You will work on large-scale data pipelines, real-time serving, and experimentation platforms.

• Design end-to-end machine learning architectures including data pipelines, feature engineering, model training, real-time serving, and evaluation systems.
• Build scalable personalization and relevance systems processing high-volume sequential behavioral data in real time.
• Develop high-velocity model training pipelines enabling rapid experimentation with production-grade rigor.
• Design and operate low-latency model serving infrastructure with seamless version rollouts.
• Create experimentation and evaluation platforms combining offline metrics and online A/B testing.
• Establish ML engineering standards for code quality, reproducibility, monitoring, and incident response.

• MSc in Machine Learning, Computer Science, Statistics, or directly related field from KTH, Chalmers, ETH Zurich, EPFL, or equivalent.
• 3+ years industry experience designing and shipping production ML systems at scale.
• Demonstrated experience in early-stage, fast-moving technology companies.
• Deep expertise in at least two of: large-scale personalization, recommendation systems, sequential behavioral modeling, real-time ML serving, or privacy-preserving ML.
• Production software engineering discipline beyond model code.

• PhD in Machine Learning, Computer Science, or directly related field.
• Prior experience as technical lead, staff engineer, or principal engineer at high-growth company.
• Background from companies like Spotify, Klarna, Google, Meta, OpenAI, DeepMind.
• Research publications or substantive technical work in relevant areas.
• Track record of delivering complete production ML systems at a fast pace.