OKX
Staff Machine Learning Engineer
San Jose, California, United States · Staff+
Sponsorship not specified$215k-$322kDetected 6 days ago
PythonMachine LearningTensorFlowPyTorchData ScienceLLMsRAGAgentic AILLMOpsMLOpsAI OrchestrationA/B TestingResearchCommunicationCollaboration
About the role
- OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps).
- OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets.
- The data spans on-chain activity, fiat deposits and withdrawals, trading behaviour, account access patterns, device intelligence, identity information, and customer interactions-signals that very few organizations can analyze together.
Responsibilities
- Own production ML systems end to end, including feature pipelines, training workflows, model serving, decision integrations, monitoring, alerting, drift detection, retraining, and incident response.
- Partner with risk strategy and product teams to translate models into effective production controls, including approval, rejection, review, cooldown, limit adjustment, account restriction, and other risk mitigation actions.
- Develop AI-powered risk capabilities such as investigation agents, case summarization, evidence collection, review recommendations, alert triage, suspicious-entity mining, and automated decision support.
- Design, build, and deploy LLM-based agents for risk operations and investigation workflows, including case triage, evidence retrieval, transaction analysis, alert summarization, review recommendations, and automated action orchestration.
- Develop production-grade agent architectures using tool calling, retrieval-augmented generation, workflow orchestration, structured outputs, memory, guardrails, and human-in-the-loop controls.
- Build evaluation frameworks for LLM agents, measuring factual accuracy, task completion, decision consistency, latency, cost, reviewer acceptance, and operational impact.
- Experience building AI agents for fraud, risk, compliance, customer operations, cybersecurity, or other high-stakes domains is a meaningful advantage.
Requirements
- A strong understanding of LLM-agent evaluation and reliability, including hallucination control, grounding, observability, permissions, failure handling, human review, latency, and cost optimization.
Compensation
- The base salary range for this position is $214,666 to $321,999.
- The salary offered depends on a variety of factors, including job-related knowledge, skills, experience, and market location.
- In addition to the salary, a performance bonus and long-term incentives may be provided as part of the compensation package, as well as a full range of medical, financial, and/or other benefits, dependent on the position offered.
Benefits
- Significant professional experience in machine learning engineering, applied data science, or a closely related field, with a strong record of taking models from prototype to production.
- Strong Python skills and hands-on experience with machine learning frameworks such as PyTorch, TensorFlow, XGBoost, LightGBM, or scikit-learn.
- You regularly use LLM coding tools, have built AI-integrated workflows or applications, and understand both the productivity benefits and the security, reliability, and governance risks.
Company info
- At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom.
- OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps). OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets. We are safe and reliable, backed by our Proof of Reserves.
- Across our multiple offices globally, we are united by our core principles: We Before Me, Do the Right Thing, and Get Things Done. These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er.
- OKX is part of OKG, a group that brings the value of Blockchain to users around the world, through our leading products OKX, OKX Wallet, OKLink and more.
- We are safe and reliable, backed by our Proof of Reserves.
- Across our multiple offices globally, we are united by our core principles: We Before Me, Do the Right Thing, and Get Things Done.
- These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er.
- About The Opportunity
- Building machine learning systems for risk at a global crypto exchange is fundamentally different from conventional ML engineering.
- The problems are complex, adversarial, and constantly evolving.
- Models must identify emerging fraud patterns, scams, account takeovers, payment abuse, and other forms of financial risk while minimizing disruption to legitimate customers.
- Success is not measured only through offline model metrics.
- It is measured through prevented losses, improved approval rates, reduced false positives, faster investigations, and more reliable customer experiences.
- This role sits within a multidisciplinary risk team of machine learning engineers, data scientists, risk strategy specialists, analytics engineers, product managers, and operations teams.
- You will work across the full ML lifecycle-from problem formulation, feature engineering, and model development to real-time deployment, monitoring, experimentation, and continuous iteration.
- You will also help shape how AI is used across the risk organization.
- LLM-assisted development, automated model workflows, AI-powered investigations, and intelligent review agents are part of the team's daily work.
- We are looking for engineers who already use these tools effectively and can help establish safe, scalable, and production-ready AI practices.
This listing is sourced directly from OKX's careers page and normalized into a canonical job model.