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