AI Engineer - Machine Learning

Listed as a remote role based in Gothenburg, Västra Götaland County, Sweden. 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 Show more Show less Seniority level Not Applicable Employment type Full-time Job function Engineering and Information Technology Industries Engineering Services