Vacant job

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 ideal candidate has experience across the entire ML chain, including data → feature → model → deployment → monitoring. You also possess a strong sense of cost consciousness, balancing Dev, Deploy, and Run costs against 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 architecture (retrieval + generation)

· Embeddings (OpenAI / Databricks / open source)

· Prompt engineering + system prompts

· Agent design (tools, routing, memory)

· Model selection (GPT vs. open models vs. cost/performance trade-offs)

· AI architecture (RAG vs. fine-tuning vs. agents)

· Chunking

· Agentic systems and RPA

· PySpark/Apache Spark

· Event-driven systems or streaming data

· Fluent in English, both spoken and written

Experience Required:

· At least 5 years of experience building and maintaining production models and AI solutions

· University degree in Data Science, ML/AI, or Software Engineering

· 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:

· Excellent communication & presentation skills

· Ability to easily collaborate with various teams and stakeholders cross-functionally

· Flexible, open-minded, and enjoys working in an international environment

Curious, analytical, with a positive can-do mindset

Additional info:

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

Swedish Driving License B is a huge plus