Senior AI/ML Engineer (Magnet-Griffeye) at Magnet Forensics

Who We Are; What We Do; Where We’re Going   Magnet Forensics is a global leader in the development of digital investigative software that acquires, analyzes, and shares evidence from computers, smartphones, tablets, and IoT-related devices. We are continually innovating so our customers can deploy advanced and effective tools to protect their companies, communities, and countries.   Serving thousands of customers globally, our solutions are playing a crucial role in modernizing digital investigations, helping investigators fight crime, protect assets, and guard national security.   With employees based around the world, Magnet Forensics has been expanding our global presence. As a part of Magnet Forensics, you can expect to make a difference in the world, no matter what role you play. You’ll be supported through learning and development, not to mention an incredible team with unbelievable talent and integrity.   If you think you would be the right person to join our team working towards this goal, we would love to hear from you!  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.  If the compensation structure for the role includes an incentive component (i.e. most Sales roles) 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.   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 [Upgrade to PRO to see contact] 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.