Senior AI/ML Engineer: Agentic Systems & RAG

Magnetforensics; Göteborgs kommun; Hybrid; SEK 623,000 - 1,068,000; Full time

Benefits offered by this job

Generous time off policies

Competitive compensation

Volunteer opportunities

Reward and recognition programs

Employee committees & resource groups

Healthcare and retirement benefits

Job summary

Magnet Forensics is seeking a Senior AI/ML Engineer to build and operate AI systems powering digital forensics capabilities. You will drive end-to-end initiatives from training to deployment, including RAG pipelines and agentic workflows that surface insights for investigators.

You’ll collaborate with Product, UX, and other engineering teams to ensure models meet real-world constraints while advancing what's possible. This role combines research, scalable systems, and responsible AI practices.

Qualifications

• 5+ years of professional experience in ML or applied AI with production delivery.
• Depth in at least one area: model training/evaluation, agentic design, or retrieval architecture.
• Experience evaluating ML/AI systems and designing evaluation distributions.
• Comfort with large, unstructured datasets and trade-offs between quality, cost, speed, and complexity.
• Proficiency in Python and modern ML frameworks like PyTorch.
• Strong cross-functional collaboration with Engineers, Researchers, Product, and Design.
• Bachelor’s or Master’s in CS/ML or related field, or equivalent practical experience.

Responsibilities

• Own complex AI/ML initiatives from ideation through experimentation, evaluation, deployment, and handoff for integration.
• Design and prototype agentic workflows where models reason, plan, call tools, and collaborate with other systems to accomplish complex tasks.
• Train, fine-tune, or adapt models when the problem demands it, and know when a well-designed system beats a bigger model.
• Work with complex, real-world datasets, developing pre-processing, augmentation, and evaluation techniques that enhance model quality and fairness.
• Collaborate cross-functionally with other engineering teams to ensure models and systems are production-ready, observable, scalable, and meet real user needs.
• Contribute to reusable engineering infrastructure that accelerates experimentation, evaluation, and deployment.
• Embed ethical, responsible, and secure AI practices into design, evaluation, and deployment decisions, raising concerns early when they surface.
• Mentor other engineers at different levels on experimental design, evaluation methodology, and technical decision-making, helping the team to level up by establishing patterns and best practices.

Skills

Model training

Evaluation

Agentic workflows

Retrieval architecture

Python

Cross-functional collaboration

Communication

Education

Bachelor’s or Master’s in Computer Science, ML, or related field

Tools

PyTorch

Inference servers

LLM/agent frameworks

MLOps tooling

Cloud (AWS/Azure)

Job description

Magnet Forensics is seeking a Senior AI/ML Engineer to build and operate AI systems powering digital forensics capabilities. You will drive end-to-end initiatives from training to deployment, including RAG pipelines and agentic workflows that surface insights for investigators.

You’ll collaborate with Product, UX, and other engineering teams to ensure models meet real-world constraints while advancing what's possible. This role combines research, scalable systems, and responsible AI practices.