Staff Engineer, Machine Learning & Data Infrastructure (Crypto)

Location: Hybrid role in San Jose, California, United States

Salary: $214,666–$321,999 annually, plus potential performance bonus, long-term incentives, and benefits.

Company

OKX is a crypto exchange and developer of the OKX Wallet, providing access to crypto trading and decentralized applications for individuals and institutions.

What you will do

• Lead the architecture and hands-on development of ML and data infrastructure for risk and fraud prevention systems.
• Optimize model inference latency, throughput, resource utilization, and cost across the full prediction path.
• Build reusable infrastructure for model packaging, versioning, deployment, serving, monitoring, and rollback across cloud platforms and regions.
• Design and optimize offline data pipelines using Databricks, Apache Spark, and warehouse or lakehouse technologies.
• Connect offline and online feature pipelines while ensuring point-in-time correctness, consistent definitions, freshness, lineage, and reproducible backfills.
• Set technical direction, mentor engineers, and lead delivery across ML, data engineering, cloud infrastructure, security, and product teams.

Requirements

• Experience building and operating large-scale distributed systems, ML infrastructure, or data platforms with technical ownership across teams.
• Hands-on experience optimizing production model performance with measurable improvements in latency, throughput, utilization, or cost.
• Deep understanding of Databricks and Spark or comparable offline data platforms, including execution plans, distributed query optimization, storage layouts, data correctness, and workload cost management.
• Experience across the model lifecycle, including training and serving integration, validation, deployment, observability, and rollback.
• Strong Python and SQL skills plus at least one backend or systems language such as Java, Scala, Go, or C++.
• Ability to turn ambiguous requirements into clear architecture and deliver maintainable systems through hands-on engineering and technical leadership.

Nice to have

• Experience with SageMaker, Alibaba Cloud PAI/EAS, or comparable ML platforms.
• Inference optimization using ONNX Runtime, TensorRT, or Triton.
• Experience with Delta Lake, MLflow, feature stores, Kafka, Flink, and low-latency data stores.
• Experience with fraud detection or financial risk systems.
• Fluency in Mandarin.

Culture & Benefits

• Competitive total compensation package with potential performance bonus and long-term incentives.
• Learning and development programs.
• Team-building programs and company events.
• Wellness and meal allowances.
• Comprehensive healthcare schemes for employees and dependants.
• Equal employment opportunity commitment; applicants with arrest and conviction records will be considered in accordance with the San Francisco Fair Chance Ordinance.

Hiring process

• Apply through the OKX internal or external careers site.