Staff AI Engineer Model Post-Training and Alignment

Skills

About the Role

You will design, execute, and optimize post-training pipelines for large language models across data strategy, reward modeling, reinforcement learning, alignment, evaluation, and production inference. You will develop domain-specific data pipelines, train specialized models, build RLAIF systems, optimize serving, and productionize training and deployment workflows.

Requirements

• Bachelor's degree in Computer Science, AI, Machine Learning, or a related field
• 8+ years of industry experience
• Large model post-training
• Preference learning
• DPO
• GRPO
• Reinforcement learning
• Domain-specific data strategies
• Training specialized small models from scratch
• Reward modeling
• RLAIF
• vLLM
• SGLang
• Low-latency production deployment

Responsibilities

• Lead and execute LLM post-training pipelines
• Design DPO and GRPO training paradigms
• Develop domain-specific data recipes and augmentation pipelines
• Train specialized small models from scratch
• Build and refine reward models
• Design and implement RLAIF closed-loop systems
• Optimize inference efficiency and deploy models
• Evaluate model performance with benchmarks and feedback loops
• Collaborate to productionize training and deployment workflows

Benefits

• L&D programs
• Education subsidy
• Team building programs
• Company events
• Wellness allowances
• Meal allowances
• Comprehensive healthcare schemes for employees and dependants