Senior AI/ML Engineer (Magnet-Griffeye)

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

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.

The work on our team is diverse. In a given quarter, you might design an evaluation harness for an agentic system, fine-tune a model, prototype a retrieval pipeline, or partner with other engineering teams on production scaling. You'll work across the full stack (training, evaluation, RAG, agentic systems, deployment, and scale), and you'll be trusted to pick the right tool for each problem. From building AI that investigators can trust to turn months of case work into days, to figuring out what "good" looks like when users are searching for a needle in a haystack they've never seen before, our team works on challenging and meaningful problems. If these are the kinds of problems that interest you, you will feel at home on this team!

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

Qualifications

• 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 and designing representative evaluation distributions

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

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

Compensation & Benefits

The compensation range is for the primary location for which the job is posted. 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.

If the compensation structure for the role includes an incentive component, the range below represents total target compensation (TTC) (base salary + variable).

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

Magnet is proud to offer benefits such as:

• 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, than to find folks who have “been there, done that”. We want to be part of your development journey, and we’ll learn as much from you as you learn from us.

How We Work

At Magnet Forensics, we take a hybrid-flexible approach to support your productivity and work-life balance. If you’re within a comfortable travel distance to one of our offices, you’ll occasionally join us in person. How often you’ll come in depends on your department and team needs, typically ranging from weekly to monthly. These in-person moments help us build stronger connections, spark new ideas, and celebrate our successes together. Most days, you can choose what works best for you, while staying in tune with your team’s goals.

We’re excited to welcome you to our team and look forward to achieving great things together - both in the office and wherever you work best!

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.

We are committed to continuous learning and to building a diverse and inclusive workforce. 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 us for 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.