OKX is hiring a Staff AI Engineer, Model Post-Training and Alignment in Engineering — APAC. The overview below is synthesized from the employer posting on job-boards.greenhouse.io: factual requirements and scope are preserved, but prose is rewritten with editorial context. Verify details and apply via the employer link.
Who We Are
At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom. OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps).
About the Opportunity
Company context from the listing:
The listing frames the Staff AI Engineer, Model Post-Training and Alignment at OKX as a hands-on product role. 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.
What You’ll Be Doing
- Lead and execute the full post-training pipeline for large language models (LLMs), including supervised fine-tuning, preference optimization, and reinforcement learning–based methods.
- Design and implement advanced training paradigms such as DPO (Direct Preference Optimization) and GRPO (Generalized Reward Policy Optimization).
- Develop domain-specific data recipes, curation strategies, and augmentation pipelines to optimize task performance.
- Conduct post-training of specialized small models from scratch, including architecture selection, dataset construction, and optimization strategy.
- Build and refine Reward Models to support alignment and downstream optimization.
- Design and implement RLAIF (Reinforcement Learning from AI Feedback) closed-loop systems.
- Optimize inference efficiency and deploy models using low-latency serving frameworks such as vLLM and SGLang.
- Evaluate model performance using both automated benchmarks and human/AI feedback loops.
- Collaborate with research and infrastructure teams to productionize training and deployment workflows.
What We Look For In You
Experience and skills the team lists as required:
- Bachelor's in Computer Science, AI, Machine Learning, or related fields with at least 8 years of industry experience.
- Strong hands-on experience across the full post-training pipeline for large models.
- Deep familiarity with preference learning and alignment techniques, including DPO, GRPO, and RL-based post-training methodologies.
- Proven experience designing domain-specific data strategies and training methodologies.
- Experience training and post-training specialized small models from scratch.
- Solid understanding of reinforcement learning fundamentals and their application to model alignment.
- Experience deploying models in low-latency production environments using frameworks such as vLLM, SGLang, or similar.
Perks & Benefits
What the posting highlights about the offer:
- Competitive total compensation package
- L&D programs and Education subsidy for employees' growth and development
- Various team building programs and company events
- Wellness and meal allowances
- Comprehensive healthcare schemes for employees and dependants
- More that we love to tell you along the process!
Please note that Hong Kong is a group-level service hub, and OKX does not carry on a business of operating a virtual asset trading platform in Hong Kong.
Notice:
All official OKX vacancies are published on this website. While roles may appear on selected third-party platforms from time to time, information on other sites may be inaccurate or outdated.
Information collected and processed as part of the recruitment process of any job application you choose to submit is subject to OKX's Candidate Privacy Notice.
Build a standout application
For the Staff AI Engineer, Model Post-Training and Alignment at OKX, reviewers look for concise evidence over buzzwords. Mirror the language of the posting sparingly, quantify support or delivery outcomes, and show how you handled ambiguity, time-zone collaboration and user empathy. Keep your resume to impact, keep your cover note to one page, and link to artifacts — tickets resolved, docs shipped, dashboards owned — that prove you can operate in a fast-moving Web3 team. Prepare to discuss a time you turned a confusing user report into a clear fix and how you measure quality in support and operations.
Web3 hiring values reliability: on-time follow-through, clear writing, and a track record of improving runbooks and tooling. Treat the application as a work sample. For interviews, be ready to walk through how you prioritize across time zones, handle a difficult user, and decide when to escalate versus resolve directly. Show how you document decisions so the next teammate benefits.
In a distributed Web3 org, trust builds through written clarity. Use the cover note to demonstrate it.