Senior /Staff AI Engineer

The role

You will design, build, and own identity-risk and trust models to detect and prevent fraud. You will work on graph and clustering methods, behavioral risk modeling, and biometric signals in high-scale production environments.

What you will do

• Build identity trust and risk models using face, liveness, document, device, and behavioral data.
• Uncover organized fraud using entity graphs, community detection, clustering, and GNN methods.
• Develop behavioral features and sequence models to detect account takeover and bot behavior.
• Leverage biometric and multimodal signals for identity-trust inputs.
• Partner with backend, product, infrastructure, risk, and compliance teams to integrate models.
• Own metric definitions, monitoring, and drift handling.
• Experiment with graph learning, multimodal fusion, and AI agents.
• Contribute to design reviews and mentor junior engineers.

Required qualifications

• 5+ years of hands-on experience building risk-control, fraud-detection, or identity-trust models.
• Significant experience deploying ML systems into production.
• Experience applying graph mining, community detection, clustering, or GNN methods to fraud or abuse detection.
• Solid experience with behavioral feature engineering and sequence modeling.
• Strong proficiency with PyTorch and/or TensorFlow.
• Solid experience with production ML systems.
• Strong coding ability in Python and SQL.
• System-level thinking to design end-to-end solutions with high availability and low latency.
• Understanding of how fraudsters adapt and how to evaluate defenses.
• Strong communication skills for working across engineering, product, and risk teams.
• Bachelor's degree or higher in Computer Science, Electrical Engineering, or a related field.

Nice to have

• Experience in identity verification, KYC/KYB, AML, face-payment or payment risk control, or biometric security.
• Hands-on experience with face recognition, liveness/anti-spoofing, OCR, document parsing, fake-ID detection, or media forensics.
• Experience with graph databases or graph computing (e.g., Neo4j, GraphX) and real-time graph features.
• Knowledge of data and AI/ML infrastructure (e.g., Spark, Flink, Kafka, Airflow, Snowflake).
• Familiarity with MLOps best practices: monitoring, retraining, and continuous improvement of production models.
• Experience with LLMs and AI agents.
• Experience working in fast-paced technology companies or security-critical environments.
• Research publications, patents, or open-source contributions in relevant AI fields.

Benefits

• Competitive total compensation package
• L&D programs and Education subsidy
• Various team building programs and company events
• Wellness and meal allowances
• Comprehensive healthcare schemes for employees and dependants

About the company

OKX is a leading crypto exchange and the developer of OKX Wallet. It is part of OKG, a group that brings the value of Blockchain to users around the world.