Senior Data Scientist, Knowledge Layer

The role

This person will design, build, and productionize machine learning models for knowledge extraction, graph-based learning, and entity resolution. They will work on the Knowledge Layer team to extract and analyze knowledge from large-scale, unstructured data.

What you will do

• Design, build, and productionize machine learning models
• Perform knowledge extraction from unstructured data
• Perform graph-based learning and inference
• Perform entity resolution and relationship discovery
• Evaluate and leverage existing ML models and frameworks
• Partner 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

Required qualifications

• 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
• Solid software engineering fundamentals, including building and maintaining APIs or services
• Ability to translate ambiguous problem spaces into practical ML solutions
• Strong communication skills
• Comfort collaborating with engineers across disciplines
• AI fluency

Nice to have

• Experience or strong interest in knowledge graphs, graph data, or graph-based ML

Benefits

• TRM coordinates and covers approved travel, lodging, and workspace for R&D work weeks
• Accommodations available for legitimate caregiving, health, immigration, and similar constraints

Before you apply

• Review the job description carefully and highlight experience and impact that demonstrate required qualifications
• Provide thoughtful and accurate answers to application questions
• Interview process includes a case study, an AI skills assessment, and a Leadership Principles interview
• Use of AI tools during interviews is not permitted without prior approval

About the company

TRM Labs provides AI-powered intelligence solutions that help public and private sector agencies investigate and disrupt crime. The company is a Series C company headquartered in San Francisco and operates as a distributed-first company.