Senior Data Scientist, Knowledge Layer

As a Senior Data Scientist and Machine Learning Engineer at TRM Labs, you'll join the Knowledge Layer team, a group focused on extracting, structuring, and analyzing knowledge from large-scale, unstructured data. You'll bring practical ML expertise to a team that sits at the intersection of knowledge graphs, entity resolution, graph extraction, and graph analytics, playing a critical role in powering TRM's core intelligence products that help public and private sector agencies investigate and disrupt crime.

What you'll do

• Design, build, and productionize machine learning models focused on knowledge extraction from unstructured data (e.g., NER, 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 (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

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 (NER), 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 range of $180,000 to $210,000 (US-based; adjusted for other geographies)
• 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 designated location (TRM coordinates and covers approved travel, lodging, and workspace; accommodations available for caregiving, health, immigration, and similar constraints)

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, providing AI-powered intelligence solutions that help agencies investigate and disrupt crime.