Senior Machine Learning Engineer, Knowledge Layer

As a Senior Machine Learning Engineer on TRM Labs' Knowledge Layer team, you bring machine learning and data science expertise to a group focused on extracting, structuring, and analyzing knowledge from large-scale, unstructured data. This team works at the intersection of knowledge graphs, entity resolution, graph extraction, and graph analytics, powering TRM's core intelligence products used by leading agencies and businesses worldwide to investigate and disrupt crime.

What you'll do

• Design, build, and productionize machine learning models focused on knowledge extraction from unstructured data (such as Named Entity Recognition and entity linking), graph-based learning and inference, and entity resolution and relationship discovery
• 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
• Contribute to the design and evolution of knowledge graphs and ontologies
• Perform exploratory data analysis 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

What you bring

• 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
• Familiarity with NLP or information extraction techniques, such as Named Entity Recognition, text classification, or embedding-based approaches
• 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
• Strong communication skills and comfort collaborating with engineers across disciplines

What we offer

• Base salary of $180,000 to $210,000 (US only; pay adjusted for location outside the US)
• Eligibility to participate in TRM's equity plan
• Participation in in-person R&D work weeks, typically 1-2 times per quarter at a TRM hub or other designated location, with travel, lodging, and workspace coordinated and covered by TRM

About TRM Labs

TRM Labs is a Series C company with $220M in total funding, backed by Goldman Sachs, Bessemer, Y Combinator, Thoma Bravo, and others. Headquartered in San Francisco, TRM operates as a distributed-first company with hubs in Los Angeles, San Francisco, New York, Washington D.C., London, and Singapore. TRM provides AI-powered intelligence solutions that help public and private sector agencies investigate and disrupt crime.