AI Engineer - Machine Learning

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