Machine Learning Engineer

Modulai AB; Stockholms kommun; On-site; SEK 600,000 - 900,000; Full time

Job summary

A machine learning company is seeking a Machine Learning Engineer to work on diverse projects. The ideal candidate will have an MSc or Ph.D. in a quantitative field, strong Python skills, and a passion for developing scalable ML solutions. Responsibilities include analyzing problems, developing APIs, and integrating models into production systems. The role emphasizes teamwork, creativity, and continuous learning in machine learning practices. The location is in Stockholm, Sweden.

Qualifications

• Excellent understanding of ML and deep learning algorithms.
• Experience deploying ML and LLM models into production.
• Strong software development skills in Python.

Responsibilities

• Work with a broad range of ML problems.
• Analyze problems and devise solution strategies.
• Develop APIs and integrate ML models into production.

Skills

Machine Learning

Python

Software engineering best practices

Data analysis

Collaboration

Education

MSc or Ph.D. in a quantitative field

Tools

Scikit-learn

AWS

Docker

Kubernetes

CI/CD tools

Job description

Overview

Modulai works with fish, trains, clothes, money, pets, office spaces, sound sensors and much more. If there is data, we do ML (Machine Learning) on it. Our team consists of devoted ML engineers with strong track records from some of Sweden's most successful startups. We work on project basis and take end-to-end responsibility. We love ML and we think that the best way for us to expand our knowledge is to be exposed to a diversified set of challenging and fun projects.

Role

MACHINE LEARNING ENGINEER

Responsibilities

• As a member of the ML-team you will be working with a broad range of problems with one common denominator - ML will be the key ingredient. The projects could be external as well as internal - and in all cases - delivery is central.
• You will have to analyze the problem at hand, come up with a solution strategy and execute on it. This typically entails gaining an in-depth understanding of the challenge, understanding the available data and then re-formulating it as a ML problem.
• It requires openness, creativity and an eagerness to learn new methodology and exploring new terrains.
• We frequently attack these problems as a team, meaning that you will have to be able to clearly explain your reasoning and code in order to engage the rest of us.
• Developing APIs and integrating ML models into production systems.
• Ensuring scalability, monitoring, and performance optimization of deployed models.

Stack and Tools

• Python / R - standard open-source libraries
• Scikit-learn and various specialized Python and R ML libraries
• Large Language Model (LLM) frameworks such as LangChain/LlamaIndex, LangGraph, CrewAI
• Cloud platforms such as AWS, GCP, and Azure
• CI/CD: DVC, Github Actions, Sagemaker/VertexAI/AzureML
• Relational database management systems
• MLOps and LLMOps tools for model deployment and monitoring
• Software engineering best practices, including testing, version control (Git), and containerization (Docker, Kubernetes)
• Orchestration: Airflow, AWS Step functions, etc Engineering/LLM/deployment: Kubernetes, docker, terraform

Background & Skills

• MSc or Ph.D. in a quantitative field
• Excellent understanding of a broad set of ML and deep learning algorithms, including LLMs
• Strong software development skills in Python and experience with software engineering best practices
• Experience deploying ML and LLM models into production environments
• A passion for lean, clean, and maintainable code
• The desire to grow and to share insights with others

Helpful Knowledge

• Deep learning frameworks and transformer-based architectures
• LLM fine-tuning, prompt engineering, and retrieval-augmented generation (RAG)
• Data pipelining and ML/LLM infrastructure best practices
• DevOps experience, CI/CD, Kubernetes, and serverless architectures
• Experience with vector databases e.g (Pinecone, redis, and ElasticSearch) for LLM applications

About Team Modulai

At Modulai we focus 100% on solving problems with machine learning (ML). We work in teams on a project basis. We work for clients, as part of the core team in startups where we have long-time engagement as well as building our own ML products.

Learning and teamwork are central to how we work. Everyone in the team is or will soon be a full-stack ML engineer capable of scoping and developing end-to-end ML solutions. You should be able to do end-to-end machine learning products by yourself but actually, never do it because we always work in teams. If there is data, we will do ML on it!

Apply here: https://modulai.teamtailor.com/