Senior Machine Learning Engineer, Knowledge Layer

TRM Labs provides AI-powered intelligence solutions that help public and private sector agencies investigate and disrupt crime. TRM’s platforms enable investigators to trace illicit activity, build cases, and construct operating pictures of threat networks. Leading agencies and businesses worldwide rely on TRM to make the world safer and more secure.
We’re looking for a Full Stack Data Scientist / Machine Learning Engineer to join our Knowledge Layer team, a group focused on extracting, structuring, and analyzing knowledge from large-scale, unstructured data. This team sits at the intersection of knowledge graphs, entity resolution, graph extraction, and graph analytics, and plays a critical role in powering TRM’s core intelligence products.
While the team has strong backend and graph engineering expertise, we are looking for someone to be the machine learning and data science voice in the room — bringing practical ML expertise, especially in knowledge extraction and graph-based ML, to help accelerate and scale our capabilities.
This is an ideal role for someone who enjoys operating end-to-end: from model selection and experimentation, to productionizing ML systems and APIs.

Design, build, and productionize machine learning models focused on:
Knowledge extraction from unstructured data (e.g., Evaluate and leverage existing ML models and frameworks to solve real-world problems efficiently
Partner closely with backend and graph engineers to integrate ML models into production services and APIs
Perform exploratory data analysis (EDA) to inform modeling decisions and system design
Own ML components end-to-end, including experimentation, evaluation, deployment, and iteration
Help shape best practices for applied ML within the Knowledge Layer team
5+ years of experience in data science, machine learning engineering, or applied ML
~ Strong programming experience in Python
~ Hands-on experience building, training, or deploying machine learning models in production
~ Experience or strong interest in knowledge graphs, graph data, or graph-based ML
~ Solid software engineering fundamentals, including building and maintaining APIs or services
~ Ability to translate ambiguous problem spaces into practical ML solutions
~ Additionally, this role may be eligible to participate in TRM’s equity plan.
Work Week Travel:

This role requires participation in in-person R&D work weeks, typically 1–2 times per quarter, at a TRM hub or other designated location. Work weeks may be scheduled with as little as 14 days’ notice, and some roles may require additional travel based on team and cross-functional needs. TRM coordinates and covers approved travel, lodging, and workspace. Accommodations are available for legitimate caregiving, health, immigration, and similar constraints.
Application Instructions
Every application is reviewed by our Talent team.
Please also provide thoughtful and accurate answers to the application questions, as these will be used to evaluate your skills and qualifications for the role.
Applying directly is the best way to ensure your application will be considered.

Most interview processes include a case study, an AI skills assessment, and a Leadership Principles interview.
AI Fluency at TRM
AI fluency is a baseline expectation at TRM.
We believe AI meaningfully changes how top performers operate. We expect every team member to use AI to accelerate and reimagine their craft, not just automate surface tasks.
At TRM, AI fluency means you are among the top 10 percent of operators in your function in how you apply AI to:
Structure and solve problems
Improve output quality
You will be evaluated on applied AI fluency during the interview process.

We are building a safer world. If something takes months elsewhere, it will ship here in days.
Our work sits at the intersection of AI, national security, and fighting crime. As a result, the way we operate requires a high level of ownership, adaptability, collaboration, and creative problem-solving.
Priorities and targets to change quickly
A high level of personal ownership and accountability
Frequent, high-touch communication
Creative problem-solving and out-of-the-box thinking
This environment is energizing for people who enjoy building, solving hard problems, and making progress in situations that are not always fully defined. We put customers first and move with speed, focus, and adaptability. We treat every plan like an experiment – test, ship, measure, and iterate quickly.
At TRM, we care deeply about our craft. We are looking for individuals who want their work to matter, who experiment with speed and rigor, and who take pride in building a safer world for billions of people. Headquartered in San Francisco, TRM operates as a distributed-first company with hubs in Los Angeles, San Francisco, New York, Washington D.Privacy Policy and Additional Information
By submitting your application, you agree to allow TRM Labs to process your personal information in accordance with our

Privacy Policy

.
Because our hiring cycles for certain positions may span 24 to 36 months, we retain your personal information for up to 36 months from the date of your application. After that period, your data is deleted unless a different retention period is required or permitted by law.
If you are located in the European Economic Area, the United Kingdom, or another jurisdiction with applicable data protection laws, you have the right to access, correct, or request deletion of your personal data at any time before that period ends. The use of AI tools of any kind (including but not limited to notetakers, interview assistants, and real-time coaching tools such as Otter.ai, Fireflies, Fathom, Cluey, or similar) during TRM interviews is not permitted without prior approval from TRM. TRM uses its own internal tools for note-taking to ensure a consistent and confidential experience for all candidates.
We are committed to providing reasonable accommodations to applicants with disabilities, and requests can be made via this

form

.