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