Machine Learning Engineer

The role

You will work across data systems to build models for entity resolution and a knowledge graph. You will contribute to MCP servers and agentic pipelines to classify, cluster and enrich millions of entities.

What you will do

• Fine tune existing entity resolution algorithms to uncover hidden connections between people and organizations across China
• Expand knowledge graph with alternative data to map out the power structure of China
• Train, test and deploy ML models that operate on tens of millions of records daily
• Work with Product to define and implement evaluation harnesses for classical ML and agentic systems
• Build agent workflows into internal tools to improve the scale and speed of our Research team

Required qualifications

• 4+ years of experience working on clustering-type ML problems
• End-to-end machine learning model experience in production
• Significant experience with python programming
• Significant experience with SQL

Nice to have

• Experience working with Frontier/SOTA models and/or fine-tuning your own LLMs for specific tasks
• Working on problems across large, heterogeneous, messy unstructured datasets and/or with semantic search, computer vision (especially OCR), or linear optimization problems
• Experience with any of the following technologies: PySpark, Temporal, FastAPI, Scikit-learn, NumPy, Docker, Terraform, Kubernetes
• Early-stage startup experience (Series B or earlier)
• B2B SaaS experience

Benefits

• Competitive compensation including salary, equity, and rapid growth potential
• 100% company-paid Medical, Dental, and Vision coverage for employees
• FSA, HSA, and 401(k) options
• Generous paid time off plus company-wide holidays
• Pre-tax commuter benefits

About the company

WireScreen is a fast-growing Series A startup building an intelligence platform for navigating global supply chains and China-related risk. The team includes a two-time Pulitzer Prize–winning journalist and senior engineers from Google, Twitter, and Oracle.