Research Engineer, Pretraining — Remote/SF with Equity

Prime Intellect, Inc.; United States; Hybride; USD 150 000 - 350 000; Plein temps

Avantages offerts par ce poste

Remote work

In-person in SF

Visa sponsorship

Relocation assistance

Quarterly off-sites

Learning opportunities

Résumé du poste

Prime Intellect is seeking a researcher to take open models from architecture experiments through full pretraining runs in a flexible work setup. You will work across the model and training stack, making research decisions and fixing systems problems.

Responsibilities include building and extending pretraining code, evaluating architecture changes, ensuring checkpointing resilience, and running controlled ablations with data and evaluation researchers.

Qualifications

• Hands-on experience training language models and modifying training code.
• Strong Python skills and experience with PyTorch.
• Practical understanding of model architectures and optimization.

Responsabilités

• Build and extend pretraining code, evaluate architecture changes, and profile training jobs.
• Make long runs resilient with reliable checkpointing and recovery.
• Run controlled ablations and work with data and evaluation researchers.

Connaissances

Python

PyTorch

Model architectures

Research methods

Description du poste

Prime Intellect is seeking a researcher to take open models from architecture experiments through full pretraining runs in a flexible work setup. You will work across the model and training stack, making research decisions and fixing systems problems.

Responsibilities include building and extending pretraining code, evaluating architecture changes, ensuring checkpointing resilience, and running controlled ablations with data and evaluation researchers.