Senior Software Engineer, Machine Learning & Data Infrastructure
OKX is a leading global cryptocurrency exchange offering a wide range of trading services, including spot and derivatives trading, as well as Web3 solutions.
Investors
Projects
About OKX
OKX is a global cryptocurrency exchange known for its trading services and financial products. The platform offers a wide range of features, including spot and derivatives trading, staking services, and a user-friendly interface suitable for both beginners and experienced traders.
Skills
Candidate Availability
About the Role
You will build and improve ML and data infrastructure behind risk and fraud prevention systems. You will optimize model inference, build deployment and serving infrastructure, develop Databricks and Spark pipelines, work on feature pipelines and services, and own projects with limited guidance.
Requirements
- Experience building and operating distributed systems, ML infrastructure or data platforms
- Hands-on experience improving production model performance
- Strong experience with Databricks and Spark or comparable platforms
- Experience with the model lifecycle
- Strong Python and SQL skills plus one of Java, Scala, Go or C++
- Ability to work independently with limited guidance
Responsibilities
- Improve model inference latency, throughput, resource utilization and cost
- Build and maintain infrastructure for model packaging, versioning, deployment, serving, monitoring and rollback
- Develop and optimize offline data pipelines using Databricks and Spark
- Work on offline and online feature pipelines
- Build reliable feature services and data integrations for risk models
- Own projects and operational issues and collaborate across teams
- Contribute through code reviews and knowledge sharing
Benefits
- Competitive total compensation package
- L&D programs
- Team building programs and company events
- Wellness and meal allowances
- Comprehensive healthcare schemes for employees and dependants
- Performance bonus and long-term incentives may be provided
