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