Member of Technical Staff, Machine Learning

Overview

Build and improve machine learning components for production systems, covering data, training, evaluation, and inference. The role focuses on shipping, debugging, and iterating on real-world ML systems.

What you'll do

• Build and improve ML components across data, training, evaluation, and inference.
• Fine-tune and adapt models within production systems.
• Implement evaluation and testing to understand model behavior.
• Build and maintain pipelines for real-world and synthetic data.
• Debug model issues, performance problems, and production incidents.
• Ship iterative improvements informed by user feedback.
• Collaborate with ML engineers and product teams.

What you'll need

• Strong foundations in machine learning and modern neural architectures.
• Hands-on experience training, fine-tuning, or deploying ML models.
• Production-quality programming skills.
• Experience with Python.
• Experience with PyTorch or JAX.
• Ability to work through ambiguity with guidance and grow ownership.
• Bias toward shipping, iteration, and continuous improvement.

Details

• Location: Sweden.
• Production ML systems run on GPUs.
• Interviews are conducted virtually and/or onsite.