1 day ago
Base Salary
$179k - $268k/yr
Responsibilities
- Improve model inference latency, throughput, resource utilization, and cost while preserving model quality.
- Build and maintain infrastructure for model packaging, versioning, deployment, serving, monitoring, and rollback across cloud platforms and regions.
- Develop and optimize offline data pipelines using Databricks, Apache Spark, and warehouse or lakehouse technologies.
- Build and operate offline and online feature pipelines with point-in-time correctness, consistent definitions, freshness, lineage, and reproducible backfills.
- Build and maintain reliable feature services and data integrations for risk models and decision systems.
- Own projects and operational issues, communicate progress and tradeoffs, collaborate cross-functionally, and contribute through technical discussions and code reviews.
Requirements
- Experience building and operating distributed systems, ML infrastructure, or data platforms with ownership of meaningful production components.
- Hands-on experience improving production model performance, including latency, throughput, resource utilization, or cost.
- Strong experience with Databricks and Spark or comparable offline data platforms, including query optimization, storage layouts, data correctness, and workload cost management.
- Experience integrating model training and serving, validation, deployment, observability, and rollback.
- Strong Python and SQL skills plus proficiency in at least one backend or systems language such as Java, Scala, Go, or C++.
- Ability to work independently through loosely defined problems and deliver maintainable systems with limited guidance.
- Preferred experience with SageMaker, Alibaba Cloud PAI/EAS, ONNX Runtime, TensorRT, Triton, Delta Lake, MLflow, feature stores, Kafka, Flink, low-latency data stores, fraud detection, or financial risk systems.
- Fluency in Mandarin is preferred.
Benefits
- Competitive total compensation package, including 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.
Tech Stack
Categories
Data EngineeringML Engineering
About OKX
OKX builds a global cryptocurrency exchange and Web3 wallet used by retail and institutional traders for spot, derivatives, and DeFi access, with developer APIs. Its business model centers on trading and financing fees, plus wallet and on-chain services. Founded in 2017 and privately held, OKX publishes monthly proof-of-reserves and operates as part of the OKG group serving crypto markets worldwide.