Senior Machine Learning Engineer, Knowledge Layer

TRM Labs develops intelligence products that help public and private sector agencies investigate and disrupt crime. The Senior Machine Learning Engineer will build and productionize machine learning capabilities for the Knowledge Layer, including knowledge extraction, graph-based learning, and entity resolution, and integrate them into production services and APIs.

Responsibilities:

• Design, build, and productionize machine learning models focused on:
  • Knowledge extraction from unstructured data (e.g., NER, entity linking)
  • Graph-based learning and inference
  • 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

Requirements:

• 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