Staff Ai Engineer, Model Post-training And Alignment

OKX

APAC
On-site
Large model post-training pipeline
Preference learning and alignment techniques
Dpo, grpo, and rl-based post-training
OKX is seeking a Staff AI Engineer with expertise in large model post-training and alignment, focusing on optimizing model performance through advanced methodologies. The ideal candidate will have extensive experience in machine learning and reinforcement learning in production environments

Job Summary

  • This role focuses on designing, executing, and optimizing post-training pipelines to improve model performance, controllability, domain adaptation, and reasoning capabilities.
  • You will work across the full lifecycle of post-training—from data strategy and reward modeling to reinforcement learning–based optimization and production-grade inference deployment.
  • Competitive total compensation package, L&D programs, Education subsidy, Various team building programs, Wellness and meal allowances, Comprehensive healthcare schemes.

Matching Summary

Match Score: 85

OKX is seeking a Staff AI Engineer with expertise in large model post-training and alignment, focusing on optimizing model performance through advanced methodologies. The ideal candidate will have extensive experience in machine learning and reinforcement learning in production environments.

Skills & Requirements

Must-have

  • large model post-training pipeline
  • preference learning and alignment techniques
  • DPO, GRPO, and RL-based post-training
  • domain-specific data strategies
  • training specialized small models
  • low-latency production environments

Nice-to-have

  • contribute to individual freedom
  • friendly, rewarding, and diverse environment
  • collaborate with research teams

Key Requirements

  • 8 years of industry experience
  • Bachelor's in Computer Science, AI, Machine Learning
  • Proven experience designing domain-specific data strategies
  • Experience deploying models in low-latency production environments

Work Rights

Not specified

Tailored Resume

Cover Letter