Machine Learning Research Engineer

About the company

Our client is a healthcare technology company.

The role

The company is recruiting for this role on behalf of our client. Research and build machine learning systems that turn complex source documents into dependable, verifiable logic running in production. You will plan and run experiments on modern language models, shape evaluation methods, and ship the strongest approaches into a live pipeline. The work is done in close partnership with founders and subject-matter experts who review the results.

What you'll do

• Build a production pipeline that converts long-form documents into executable modules, covering parsing, transformation and checking.
• Iterate on the format used to express decision logic.
• Run comparative experiments across model families and adaptation techniques, including output-format control.
• Co-design datasets, metrics and error analysis to measure accuracy and faithfulness to sources.
• Create adversarial test cases that probe how well outputs match their sources and how complete and correct they are.
• Turn model failures into hypotheses, experiments and stronger system design.
• Write rule-based automated checks that rely on program analysis and logical reasoning tools to expose unclear, conflicting or missing information.
• Work directly with founders and expert reviewers who validate the outputs.

Requirements

What we're looking for

• Experience shipping machine learning or NLP systems to production, beyond research prototypes.
• Background in language tooling, automated code generation or rigorous verification techniques.
• Production-level Python and PyTorch skills, along with the Transformers ecosystem.
• Hands-on work evaluating, training or prompting large language models.
• Familiarity with logical solvers, intermediate representation design or domain-specific languages.
• Bachelor's degree or higher in computer science, machine learning, math or a related technical field.
• Clear written communication about failed experiments and uncertainty.
• Authorization to work in the US or Canada.
• Willingness to attend in-person team offsites several times a year.

Bonus points

• Experience at a leading AI research lab.
• PhD in computer science, machine learning, NLP or statistics.
• Prior work in an environment where safety and compliance matter.

Benefits

Compensation and benefits

• Base salary: USD 250,000 to 400,000 per year
• Equity
• Health, dental and vision coverage
• Flexible time off
• Paid parental leave

Location and work model

• Remote (US or Canada)
• Full-time