Senior Machine Learning Engineer

About

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 Machine Learning Engineer to join our team, designing, experimenting with, and optimizing applied ML and AI systems that power our digital forensics capabilities. You will lead the development of new models, training techniques, evaluation methods, and AI‑powered systems that surface critical leads and insights for investigators, helping them solve cases faster and with greater confidence.

As part of this team, you’ll work closely with Product, UX, and our Brain team to ensure our models and systems advance what’s possible while meeting real‑world constraints. You’ll own complex initiatives end‑to‑end, including ideation and experimentation, to evaluation and handoff for integration, working with our team to advance the state‑of‑the‑art in digital forensics.

What You’ll Do

• Design, implement, and evaluate state‑of‑the‑art ML/AI models and systems;
• Lead experiments, define success metrics, build evaluations, and iterate to improve performance, efficiency, and reliability;
• Collect, build, and work with complex, real‑world datasets, developing preprocessing, augmentation, and feature engineering techniques that enhance model training and fairness;
• Design and prototype agentic workflows where models reason, plan, call tools, and collaborate with other systems to accomplish complex tasks;
• Collaborate cross‑functionally with our Brain team to ensure models are production‑ready, observable, scalable, and meet real user needs;
• Stay at the forefront of ML/AI research, assessing new techniques, frameworks, and trends, and translating them into practical innovations for our products;
• Contribute to building reusable research infrastructure and tooling that accelerates experimentation and improves reproducibility;
• Ensure ethical, responsible, and secure AI practices are integrated into model design, training, and evaluation;
• Mentor other engineers on ML and AI best practices, experimental design, evaluation methodology, and technical decision‑making.

What We’re Looking For

• 5+ years of professional experience in machine learning or applied AI, with a track record of delivering models into production or production‑ready pipelines;
• Strong Python programming skills, with experience in building maintainable, scalable ML systems;
• Experience designing and running experiments, selecting appropriate metrics, and evaluating models;
• Practical experience working with large language models in production or research prototypes, including prompt engineering, fine‑tuning or adaptation, and/or retrieval‑augmented generation;
• Hands‑on experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and deployment frameworks (e.g., Triton, Torch Server);
• Experience working with large, complex, and/or unstructured datasets, with a strong understanding of trade‑offs between model quality, cost, inference speed, and system complexity;
• Ability to work cross‑functionally with engineers, researchers, product managers, and designers;
• Strong communication skills for both technical and non‑technical audiences;
• Bachelor’s or Master’s degree in Computer Science, Machine Learning, or a related field.