Research Engineer - Pretraining Data

Prime Intellect AI; San Francisco (CA); Remote; USD 150.000 - 350.000; Vollzeit

Benefits dieser Stelle

Flexible work arrangements

Visa sponsorship & relocation

Team off-sites & learning

Zusammenfassung

Prime Intellect AI is assembling the open frontier AI platform. We seek a data-focused role to source and curate training data, turning sources into training-ready corpora, and collaborating with researchers to test data efficacy for pretraining and midtraining.

You will build scalable pipelines, ensure data quality, provenance, and reproducibility, and help design data mixtures for long-context reasoning and coding tasks. Remote or SF-office options available.

Qualifikationen

• Strong Python programming skills and experience building data-processing systems.
• Experience with large datasets and storage/compute tradeoffs.
• Ability to translate research goals into concrete data quality checks.
• Comfort delivering end-to-end data pipelines with researchers.
• Experience with web crawling, document extraction, or corpus curation.

Aufgaben

• Find, evaluate, and collect valuable training data from web, code, and technical sources.
• Build scalable pipelines that convert raw data into training-ready corpora.
• Develop checks for data quality, coverage, and contamination.
• Collaborate with researchers on data mixtures and sampling strategies.
• Prepare data for long-context learning and testing with synthetic data.
• Run experiments and use evaluations to decide data sources and processing.
• Ensure data processing is reproducible with clear provenance and versioning.

Kenntnisse

Python programming

Data pipelines

Distributed systems

Data quality checks

Tools

Spark

Ray

Git

Jobbeschreibung

Own Your Intelligence

Prime Intellect is building the open superintelligence stack: the infrastructure frontier AI labs build internally, made available to every ambitious AI team.

Our platform, Lab, unifies compute, environments, evaluations, secure sandboxes, high-performance training, and deployment into one full-stack system for post‑training at frontier scale - from SFT and RL to tool use, agent workflows, and continuously improving production models. We are building open frontier AI: open‑source models trained end to end for long‑horizon tasks like autonomous research, and the full‑stack platform our own research team uses to build them. The next generation of AI companies, enterprises, and research teams do not just need more GPUs. They need the ability to turn their own workflows, tools, data, and feedback loops into superintelligence they own.

Prime Intellect has raised $150M in total funding from Founders Fund, Radical Ventures, NVIDIA, and exceptional AI, infrastructure, and enterprise operators, including Andrej Karpathy, Dwarkesh Patel, and leaders and founders from Ramp, Perplexity, Harvey, Mercor, Zapier, Datadog, Cognition, OpenAI, Thinking Machines, Together AI, SemiAnalysis, LangChain, Browserbase, Cloudflare, Sierra, Databricks, Airbnb, OpenRouter, Standard Intelligence, Fleet, Core Auto, and more. We are looking for people who want to build at the intersection of frontier research, real infrastructure, and go‑to‑market for a category that does not fully exist yet.

Role Impact

Build the data behind Prime Intellect's next generation of open models. You’ll find valuable sources, turn them into training‑ready corpora, and work with researchers to test what actually improves the model. Your work will shape both broad pretraining and the data used to develop reasoning, coding, and long‑context capabilities.

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.

Requirements

• Strong programming skills in Python and a record of building reliable data‑processing or distributed systems.

• Experience working with large datasets and managing the practical tradeoffs of 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.

Relevant experience may include:

• Web crawling, document extraction, code‑data collection, or large‑scale corpus curation.

• Deduplication, text classification, quality scoring, synthetic data, or training‑data contamination analysis.

• Data‑mixture experiments, curriculum learning, or measuring how dataset changes affect model performance.

Nice to have

• Experience preparing datasets for language‑model pretraining, continued pretraining, or midtraining.

• Work with distributed processing frameworks such as Spark, Ray, or comparable systems.

• Experience with multilingual, mathematical, scientific, or repository‑level code data.

• Familiarity with tokenization, training data loaders, and the needs of large model‑training runs.

Depth in data acquisition, scalable processing, or data research can be a strong starting point. We don’t expect every candidate to have worked across all of them.

Benefits and Perks

• Cash Compensation Range of $150-350k, including equity incentives, aligning your success with the growth and impact of Prime Intellect.

• Flexible work arrangements, with the option to work remotely or in‑person at our offices in San Francisco.

• Visa sponsorship and relocation assistance for international candidates.

• Quarterly team off‑sites, hackathons, conferences and learning opportunities.

• Opportunity to work with a talented, hard‑working and mission‑driven team, united by a shared passion for leveraging technology to accelerate science and AI.