L

LawZero

Nonprofit research on AI safety systems

Senior ML Data Processing Developer

Data EngineeringUpdated on Nov 12
Full-timeNo salary listedIn-person

About the Job

Requirements

  • A degree in computer science, software engineering, or a related field.
  • At least 5 years of experience in data processing, machine learning engineering, or Natural Language Processing.
  • A proven track record handling massive unstructured text datasets at trillion-token scale.
  • Hands-on experience with distributed processing frameworks such as Spark, Ray, or Flink, including designing and optimizing high-throughput pipelines.
  • Experience implementing data privacy measures such as personally identifiable information scrubbing, content-safety filtering for toxicity and bias, and evaluation-contamination prevention.
  • Ability to work across Research, Engineering, and/or Legal/Governance teams and translate varied requirements into concrete pipeline work.
  • Strong Python proficiency, including experience writing production-grade data-processing code.
  • Experience with pipeline orchestration frameworks such as Airflow, Prefect, or Dagster.

Responsibilities

  • Partner with the Research team to define, build, automate, scale, and manage data pipelines that transform raw web-scale data into training datasets for the Scientist AI.
  • Build and maintain data processing pipelines for deduplication, model-based quality scoring, heuristic filtering, toxicity removal, personally identifiable information scrubbing, metadata extraction, proprietary data transformations, dataset versioning, and provenance tracking, optimizing throughput and cost at scale.
  • Ensure ingested data meets compliance requirements, internal Data Governance policies, and legal obligations.
  • Develop and refine the scoring and filtering toolchain, including heuristics, LLM-as-a-judge evaluators, machine-learning classifiers, metadata extraction modules, and human-in-the-loop review workflows for data processing and quality assurance.
  • Instrument data processing pipelines with data-quality monitoring, guardrails, and alerting to catch regressions before they propagate downstream.
  • Collaborate with the Research team and other teams to understand evolving data requirements, identify and acquire large-scale text corpora, conduct systematic coverage analyses to identify corpus gaps, develop targeted acquisition strategies, and work with the Legal & Governance Team to license new data sources.
  • Design and maintain leakage detection mechanisms to guard against evaluation contamination across all stages of the data processing pipeline.
  • Build internal tooling and interfaces that enable researchers to explore, query, and understand available datasets.

Desired Qualifications

  • Experience training, fine-tuning, or deploying machine-learning models for data-quality tasks, including classifiers and LLM-based evaluators, and familiarity with LLM inference optimization tools such as vLLM or SGLang.
  • Familiarity with containerized deployment using Docker and Kubernetes, and infrastructure-as-code practices.
  • Familiarity with machine-learning experiment tracking tools such as Weights and Biases.
  • Experience with data licensing workflows or web-scale data acquisition.
  • Contributions to open-source data processing or Natural Language Processing tooling.

About the Company

LawZero

Nonprofit research on AI safety systems

LawZero advances AI safety through research and technical development aimed at building AI systems that are safe by design and prioritize public welfare. Operating as a nonprofit under the Mila Quebec AI Institute, it conducts research and develops safety methods, evaluation tools, and frameworks that mitigate catastrophic AI risks. Its work is funded by generous donors, including Jaan Tallinn and the Survival and Flourishing DAF, which provides market-insulated support to focus on long-term safety goals rather than commercial pressures. LawZero's outputs are research results and practical safety solutions that can be used to guide the development of AI for global benefit. Compared with for-profit or consumer-focused rivals, LawZero differentiates itself through nonprofit funding, a close collaboration with Mila, and a mission-centered emphasis on human flourishing and public welfare. The ultimate goal is to reduce existential AI risks and ensure AI benefits humanity at large.

Company size
11-50
Company stage
Grant
Total funding
$250M
Headquarters
Montreal, Canada
Founded
2025

Recently Posted Jobs

Explore more opportunities at LawZero

See more jobs