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