Senior Machine Learning Engineer (AI Agent)
Description
We're hiring a Senior NLP Engineer to own the accuracy of our information-extraction pipeline. You'll work on hard problems extracting material properties and quantitative measurements from long, complex documents — patents and scientific literature where the signal is scattered across sections and relationships span paragraphs. This is a hands-on senior role with end-to-end ownership: you'll independently scope, prototype, and ship algorithmic improvements, define how we measure quality, and raise the extraction precision bar to production grade.
Responsibilities
-
Responsible for analysing and understanding of the massive structured and unstructured data by NLP tasks, such as data processing, NER, relationship extraction, NOR etc.
-
Own and improve our information-extraction pipeline, from model to production
-
Solve document-level extraction at scale — design how we handle long documents where entities, properties, and their relationships span across sections
-
Independently drive algorithm optimization, develop and fine-tune models to meet business requirements.
-
Build hybrid systems combining NER, rule-based methods, and LLMs, applying each where it genuinely wins
-
Define evaluation methodology / annotation strategy / quality metrics with domain experts
-
Process and extract from large-scale corpora reliably and efficiently
Qualifications
-
Master or Bachelor degree in Computer Science or related field.
-
4+ years of hands-on NLP/ML engineering in production, owning model or pipeline quality
-
Proven experience with long-document / document-level information extraction and normalization
-
Solid engineering fundamentals from working at scale
-
Strong in NER, entity methodology; across rule-based, statistical, and LLM-based methods
-
Experience in vibe coding and solid coding skills.
Skills
As published by lever · 4 questions · 2 written answers
Basics
Resume/CV, Full name, Email, Phone, Current location, Current company, LinkedIn URL, Twitter URL, GitHub URL, Portfolio URL, Other website
Pick from a list (2)
- Which of the following best describes about your status? optional
- Which of the following best describes about you ability to communicate in Mandarin optional
Written answers (2)
- When are you available to start?
- What are your salary expectations (base and total earnings)?

SoftBank Vision Fund