AI Engineer - Machine Learning

Integro Consulting AB; Göteborgs kommun; On-site; SEK 900,000 - 1,200,000; Full time

Job summary

Integro Consulting AB in Sweden seeks a Senior Machine Learning Engineer to enhance our client\'s data platform (Databricks) with AI capabilities across the full ML lifecycle—from data features to deployment and monitoring. You will work with cost-conscious Dev-Deploy-Run practices and help shape AI solutions in an international setting.

Strong Python, Azure AI stack, and Spark/PySpark experience are required, along with hands-on work with TensorFlow, PyTorch, MLlib and Scikit-Learn.

Qualifications

• University degree in data science/ML/AI or software.
• Hands-on Azure and Databricks experience.
• Co-pilot experience and ability to connect agents.
• Experience with RAG systems and embeddings.
• Mentor junior ML and data engineers.

Responsibilities

• Contribute to all parts of an AI solution.
• Design, deploy and monitor production ML models.
• Collaborate with cross-functional teams and stakeholders.
• Mentor junior ML and data engineers.

Skills

End-to-end ML lifecycle

Python

Azure AI stack

Prompt engineering

Cost-conscious Dev-Deploy-Run

Communication

English fluent

Agent integration

RAG concepts

Model selection tradeoffs

Event-driven systems

Education

University degree in Data Science/ML/AI/SW

Tools

Azure Databricks

PySpark

TensorFlow

PyTorch

MLlib

Scikit-Learn

OpenAI embeddings

RAG architectures

Agent design

Spark streaming

Job description

Job Description: We are looking for a Senior Machine Learning Engineer to enhance our client's data platform (Databricks) with AI capabilities. The person we are seeking has experience from the whole ML chain including data feature model deployment monitoring. You also have experience from cost consciousness – Dev, Deploy and Run vs. business value.

AI Engineer

Job Description: We are looking for a Senior Machine Learning Engineer to enhance our client's data platform (Databricks) with AI capabilities. The person we are seeking has experience from the whole ML chain including data feature model deployment monitoring. You also have experience from cost consciousness – Dev, Deploy and Run vs. business value.

Skills required

• Able to contribute to all parts of an AI solution
• Azure AI stack such as Azure AI services, Microsoft Foundry etc
• Strong Python skills
• Proficient in ML frameworks such as Tensorflow, PyTorch, MLlib and Scikit-Learn
• RAG-arkitektur (retrieval + generation)
• Embeddings (OpenAI / Databricks / open source)
• Prompt engineering + system prompts
• Agent-design (tools, routing, memory)
• Model selection (GPT vs open models vs cost/perf tradeoffs)
• AI-achitecture (RAG vs fine-tuning vs agents)
• Chunking
• Agentic systems and RPA
• PySpark/Apache Spark
• Event driven systems or streaming data
• You are fluent in English both spoken and written

Experience Required

• At least 5 years of experience building and maintaining production models and AI solutions
• You have a university degree within Data Science, ML/AI, SW
• Hands on experience in designing and operating functions in Azure and Databricks
• Co-pilot experience
• Experience in connecting different agents
• Experience with RAG systems
• Experience mentoring junior ML and Data engineers

Soft Skills

• You have excellent communication & presentation skills
• You can easily collaborate with various teams and stakeholders cross-functionally
• You are flexible, open and enjoy working in international environment

Curious, analytical, with a positive can-do ment

Additional info

You are located in Sweden and can work 100% onsite in Gothenburg

Swedish Driving License B is a huge plus