AI Research Engineer

About the role

As a founding post-training specialist, you will own the end-to-end alignment and reasoning stack taking base pre-trained models through advanced alignment. You will design and execute novel post-training recipes spanning RLAIF, Constitutional AI, and dense process supervision (PRMs) to unlock frontier reasoning and domain alignment directly on raw base weights.

What you'll do

• Process Reward Modeling (PRMs): Construct and train step-level PRMs to evaluate granular reasoning steps, intermediate chain-of-thought derivations
• Verifiable Reward & Environment Design: Implement deterministic verifiers, execution sandboxes for code and math
• Own the end-to-end alignment and reasoning stack taking base pre-trained models through advanced alignment

What you need

• 3–6 yrs of experience
• Research
• Data structures
• LLMs
• PyTorch
• Any graduate