Senior AI/ML Engineer (Magnet-Griffeye)

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

Benefits offered by this job

Generous time off policies

Competitive compensation

Healthcare benefits

Volunteer opportunities

Job summary

Magnet Forensics in Sweden is seeking a Senior AI/ML Engineer to lead initiatives in building and operating AI systems that enhance digital forensics capabilities. The role involves training models, collaborating across teams, and mentoring junior engineers. The ideal candidate should have over 5 years of experience in ML, a strong background in AI systems, and proficiency in Python and modern AI frameworks. This role offers a competitive salary and generous benefits including time off and healthcare support.

Qualifications

• 5+ years of experience in ML or applied AI, with a track record of production models.
• Experience with large and complex datasets, understanding trade-offs in model quality and system complexity.
• Strong communication skills, capable of working with cross-functional teams.

Responsibilities

• Own complex AI/ML initiatives from ideation to integration.
• Design and prototype agentic workflows.
• Develop evaluation techniques for ML models.

Skills

Model training and evaluation

Agentic system design

Proficiency in Python

Cross-functional collaboration

Education

Bachelor's or Master’s in Computer Science or related field

Tools

PyTorch

MLOps tooling

Job description

Role Overview

We are looking for a Senior AI/ML Engineer to join our team, building and operating AI systems that power our digital forensics capabilities; our work ranges from model training and evaluation through deployment, retrieval‑augmented generation (RAG) pipelines, and agentic workflows. You will own complex initiatives end‑to‑end, shipping AI‑powered systems that surface critical leads and insights for investigators, helping them solve cases faster and with greater confidence. You'll work closely with Product, UX, and other engineering teams to ensure our models and systems advance what's possible while meeting real‑world constraints.

What You’ll Do

• 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.

What We’re Looking For

• 5+ years of professional experience in ML or applied AI (or equivalent depth demonstrated through delivered work), with a track record of delivering models or AI systems into production.
• Demonstrated depth in at least one of: model training and evaluation, agentic system design, or retrieval and evaluation architecture, with working familiarity across the other areas.
• Experience evaluating ML/AI systems. Designing representative evaluation distributions, checking that training signals or metrics reflect the actual outcome you care about.
• Comfortable working with large, complex, and/or unstructured datasets, with a strong understanding of trade‑offs between model quality, cost, inference speed, and system complexity.
• Proficiency in Python and working fluency with modern ML/AI frameworks and tooling (e.g., PyTorch, inference servers, LLM/agent frameworks).
• Strong communication and cross‑functional collaboration skills; comfortable working with Engineers, Researchers, Product, and Design.
• Bachelor’s or Master’s degree in Computer Science, Machine Learning, or a related technical field, or equivalent practical experience.

Nice to Have Skills

• Familiarity with vector databases, embedding models, and context retrieval strategies.
• Background in NLP, computer vision, or other relevant ML domains.
• Familiarity with MLOps tooling (e.g., experiment tracking, model versioning, CI/CD for ML).
• Contributions to open‑source ML/AI projects or publications in peer‑reviewed venues.
• Experience working with cloud providers like AWS or Azure; or other relevant production AI/ML infrastructure.
• Experience working with AI tools as part of your development workflow (e.g., Claude, GitHub Copilot, etc.).

Compensation & Benefits

The Compensation range is for the primary location for which the job is posted. Please note that the actual compensation may vary depending on location and job‑related factors such as qualifications, experience, knowledge and skills. If you are applying for this role outside of the primary location and you are selected for an interview, the Talent Acquisition Partner can share more information with you.

MIN: 623,000 - MID: 890,000 - MAX: 1,068,000 SEK a year

• Generous time off policies
• Competitive compensation
• Volunteer opportunities
• Reward and recognition programs
• Employee committees & resource groups
• Healthcare and retirement benefits

Indicators of Success

We’re looking for someone who checks off most, but not all, of the boxes listed in “skills and experiences”. It’s more important to us to find candidates who can display indicators of success through skills they have developed and experiences they have been a part of, rather than to find folks who have ‘been there, done that’.

How We Work

At Magnet Forensics, we take a hybrid‑flexible approach to support your productivity and work‑life balance. You can choose what works best for you while staying in tune with your team’s goals.

The Most Important Thing

We’re looking for candidates that can provide examples of how they have demonstrated Magnet CODE in their previous experiences.

CARE – We care about each other and our mission to make a difference in the world.

OWN – We are accountable for our results – while never forgetting to act with integrity, empathy, and respect.

DEDICATE – We put our heart and soul into meeting the needs of our customers and helping them serve the people they protect.

EVOLVE – We are constantly innovating and exploring new ways to work together to make an impact with our work.

Here at Magnet Forensics, we are committed to continuous learning and are focused on building a diverse and inclusive workforce. This commitment will be reflected in our hiring processes and embedded in our values and how we treat one another. If you’re interested in this role, but do not meet all of the qualifications listed above, we encourage you to apply anyways.

Magnet Forensics is an Equal Opportunity Employer and considers applicants for employment without regard to race, colour, religion, sex, orientation, national origin, age, disability, genetics or any other basis forbidden under federal, provincial, or local law. We are committed to providing an inclusive, accessible recruitment process and work environment. Accommodation is available to all applicants upon request throughout the hiring process. Please contact [e-mail] should you require any accommodations.

All offers of employment at Magnet are contingent upon satisfactory completion of a background check. All background checks will be conducted in accordance with all applicable laws. Magnet will consider each position’s job duties, among other factors, in determining what constitutes satisfactory completion of the background check. Refusal to consent to a background check may be grounds for revoking an offer of employment.

US Applicants: Magnet Forensics participates in E‑Verify and will provide the federal government with your Form I‑9 information to confirm that you are authorized to work in the U.S.

Magnet Forensics handles and uses personal data of job applicants in line with its Recruitment Privacy Policy found here.