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