P

Prime Intellect

Open frontier-model stack for enterprise RL

Research Engineer - Pre-training

AI ResearchUpdated on Nov 13
Full-time

About the Job

Requirements

  • Hands-on experience training language models, including modifying training code and tracking the effects through actual runs.
  • Strong Python skills and experience with PyTorch or a comparable framework, with enough systems understanding to debug distributed training.
  • A practical understanding of model architectures and optimization, including their effect on convergence.
  • Ability to design an ablation that answers a research question and explain what its results do and do not show.

Responsibilities

  • Build and extend the pre-training code in prime-rl, taking small experiments into full training runs.
  • Evaluate model architecture changes, including mixture-of-experts and attention designs, against model quality and training cost.
  • Profile training jobs and improve end-to-end throughput, working with systems engineers on hardware-aware execution.
  • Make long runs resilient to failures, with reliable checkpointing and recovery.
  • Investigate training dynamics and numerical stability, checking that optimizations preserve convergence.
  • Run controlled ablations to guide larger experiments, keeping the setup and results reproducible.
  • Work with data and evaluation researchers on training mixtures, including midtraining and long-context experiments.

About the Company

Prime Intellect

Open frontier-model stack for enterprise RL

Prime Intellect provides an open, end-to-end superintelligence stack for enterprises to train and continually improve their own AI agents. The platform includes scalable compute, large-scale reinforcement learning, environments, sandboxes, evaluations, and deployment tools, and it ships open frontier models to customers. It covers the full lifecycle from training to deployment plus post-training support, enabling agentic training at scale. Its differentiator is the open, end-to-end stack used across thousands of customers with strong backing from top investors, and its goal is to help organizations build and iteratively improve their AI agents using open frontier models.

Company size
51-200
Company stage
Series A
Total funding
$150.5M
Headquarters
San Francisco, CA
Founded
2024

Recently Posted Jobs

Explore more opportunities at Prime Intellect

See more jobs