Senior /Staff AI Engineer

As a Senior AI Engineer at OKX (part of OKG), you will design, build, and own key identity-risk and trust models that protect millions of users on a leading global crypto exchange. You join the security and compliance team on the front lines of OKX's mission to detect and prevent identity fraud, account takeover, and large-scale organized abuse in real time. This is an end-to-end, research-to-production role where you own algorithms in your domain and ship models into real-money, high-risk, large-scale production environments.

What you'll do

• Build identity trust and risk models that fuse face, liveness, and document verification scores with device, IP, geo, and behavioral data for high-concurrency real-time decisioning.
• Uncover organized fraud with graph and clustering methods, 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 detect account takeover, bot and script behavior, and abnormal sessions.
• Leverage biometric and multimodal signals, turning face embeddings, liveness and anti-spoofing results, document-forgery and DeepFake signals into reliable identity-trust inputs, including biometric similarity for cross-account linkage and duplicate-identity detection.
• Drive execution and collaboration, delivering measurable results while managing ambiguity; partner with backend, product, infrastructure, risk, and compliance teams to integrate models into the production decision chain, owning metric definitions, the balance between false positives and missed fraud, pass rate, monitoring, and drift handling.
• Advance applied research and development, experimenting with graph learning, multimodal fusion, and AI agents; stay current with academic and industry advances and translate them into practical defenses.
• Raise the technical bar of the team through 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 and ML expertise, with strong proficiency in PyTorch and/or TensorFlow, and solid experience with production ML systems.
• Strong coding ability in Python, SQL, and other languages, with system-level thinking to design end-to-end solutions with high availability and low latency.
• An adversarial mindset, understanding 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.
• 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, payment risk control, or biometric security.
• Hands-on experience with face recognition, liveness and anti-spoofing, OCR, document parsing, fake-ID detection, or media forensics.
• Experience with graph databases or graph computing (Neo4j, GraphX) and real-time graph features.
• Knowledge of data and AI/ML infrastructure (Spark, Flink, Kafka, Airflow, 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
• Learning and development programs and education subsidy for your growth and development
• Team building programs and company events
• Wellness and meal allowances
• Comprehensive healthcare schemes for employees and dependants

About OKG

OKX is a leading crypto exchange and developer of OKX Wallet, trusted by millions for crypto trading and decentralized applications, and by hundreds of large institutions seeking access to crypto markets. OKX is part of OKG, a blockchain group bringing crypto value to users worldwide through products including OKX, OKX Wallet, and OKLink.