Senior /Staff AI Engineer

As a Senior/Staff AI Engineer at OKX, you sit on the front lines of the exchange's security and compliance mission. You will design, build, and own key identity-risk and trust models that detect and prevent identity fraud, account takeover, and large-scale organized abuse in real time at the scale of a global financial platform. This is an end-to-end, research-to-production role where you combine identity verification results, device and geo data, and user behavior into trust and risk signals, and use graph and clustering methods to uncover organized, industrialized fraud.

What you'll do

• Build identity trust and risk models by designing and shipping coarse-to-fine risk models that fuse face, liveness, and document verification scores with device, IP, geo signals, and behavioral data for high-concurrency real-time decisioning.
• Uncover organized fraud with graph and clustering methods by building entity graphs across accounts, devices, faces, documents and templates, and applying community detection, clustering, and GNN methods to detect fraud rings and large-scale abuse.
• Develop behavioral features and sequence models to engineer real-time and offline behavioral features, and build sequence models to detect account takeover, bot and script behavior, and abnormal sessions.
• Leverage biometric and multimodal signals by turning biometric features such as face embeddings, face quality, liveness and anti-spoofing results into reliable identity-trust inputs for risk decisions, including biometric similarity for cross-account linkage and duplicate-identity detection.
• Drive execution and collaboration by delivering measurable results while managing ambiguity, partnering with backend, product, infrastructure, risk, and compliance teams to integrate models into the production decision chain, and owning metric definitions, false-positive and missed-fraud balance, pass rate, monitoring, and drift handling.
• Advance applied R&D by experimenting with graph learning, multimodal fusion, and AI agents, and staying current with academic and industry advances to translate them into practical defenses.
• Raise the technical bar by contributing to design reviews, sharing applied-AI best practices, and mentoring junior engineers.

What you bring

• 5+ years of hands-on experience building risk-control, fraud-detection, or identity-trust models, including significant experience deploying ML systems into production.
• Graph and clustering expertise, with experience applying graph mining, community detection, clustering, or GNN methods to fraud or abuse detection.
• Solid experience with behavioral feature engineering and sequence modeling, including real-time feature pipelines.
• Deep AI/ML expertise with strong proficiency in PyTorch and/or TensorFlow, and solid experience with production ML systems.
• Strong coding ability in Python, SQL, or similar languages, and system-level thinking to design end-to-end solutions with high availability and low latency.
• An adversarial mindset, with understanding of how fraudsters adapt and how to evaluate and iterate defenses against evolving attacks.
• Strong communication skills and experience working across engineering, product, and risk teams.
• A 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 such as Neo4j or GraphX, and real-time graph features.
• Knowledge of data and AI/ML infrastructure such as Spark, Flink, Kafka, Airflow, or Snowflake.
• Familiarity with MLOps best practices including monitoring, retraining, and continuous improvement of production models.
• Experience with LLMs and AI agents.
• Research publications, patents, or open-source contributions in relevant AI fields.

What we offer

• Competitive total compensation package
• L&D programs and education subsidy for your growth and development
• Various team-building programs and company events
• Wellness and meal allowances
• Comprehensive healthcare schemes for you and your dependents

About OKX

OKX is a leading crypto exchange and developer of OKX Wallet, giving millions access to crypto trading and decentralized applications. OKX is trusted by hundreds of large institutions seeking access to crypto markets and is backed by its Proof of Reserves. OKX is part of OKG, a group that brings the value of blockchain to users worldwide through products including OKX, OKX Wallet, and OKLink.