Pretraining Data Engineer for Open Models

OpenKrill; United States; Hybrid; USD 150,000 - 350,000; Full time

Benefits offered by this job

Remote or SF office

Visa sponsorship

Relocation assistance

Team off-sites

Hackathons

Learning opportunities

Equity incentives

Job summary

Prime Intellect is building an open frontier AI data platform. You will build training data corpora, evaluate data quality, and collaborate with researchers to shape model pretraining and long-context capabilities.

This role focuses on scalable data pipelines, data extraction, and ensuring clean, provenance-backed datasets for frontier AI models. You will work across data sources, implement robust processing, and run experiments to optimize data mixtures and sourcing strategies, supporting both

Qualifications

• Strong programming skills in Python and a record of building reliable data-processing or distributed systems.
• Experience working with large datasets and managing storage, compute, throughput, and reproducibility.
• Ability to diagnose data-quality issues and translate research goals into measurable checks and experiments.
• Comfort owning an evolving technical problem end to end and working closely with researchers to test assumptions and improve the system.

Responsibilities

• Find, evaluate, and collect valuable training data from web, code, and technical sources.
• Build scalable pipelines that turn raw data into training-ready corpora through extraction, cleaning, deduplication, and filtering.
• Develop checks for data quality, coverage, and contamination, and use them to improve collection and processing.
• Work with researchers on data mixtures and sampling strategies for pretraining and midtraining.
• Prepare and evaluate data for long-context learning, reasoning, and coding, testing synthetic data where it proves useful.
• Run controlled data experiments and use model evaluations to decide which sources, mixtures, and processing choices are worth scaling.
• Make large-scale data processing efficient, reproducible, and recoverable, with clear source provenance and dataset versioning.

Skills

Python

Data processing

Distributed systems

Data quality checks

Tools

Spark

Ray

SQL

Job description

Prime Intellect is building an open frontier AI data platform. You will build training data corpora, evaluate data quality, and collaborate with researchers to shape model pretraining and long-context capabilities.

This role focuses on scalable data pipelines, data extraction, and ensuring clean, provenance-backed datasets for frontier AI models. You will work across data sources, implement robust processing, and run experiments to optimize data mixtures and sourcing strategies, supporting both