Staff Machine Learning Engineer
Okx · San Jose, California, United States · 2026-07-16
About this role
<h3 class="heading-2 ace-line old-record-id-doxuslsyQOGHoiYb47TiA1n51Th"><strong>Who We Are</strong></h3> <p>At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom.</p> <p>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.&nbsp;</p> <p>Across our multiple offices globally, we are united by our core principles: <em>We Before Me</em>, <em>Do the Right Thing</em>, and <em>Get Things Done</em>. These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er.</p> <p>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.</p> <h3 class="heading-2 ace-line old-record-id-doxus9qcafz8J9vhi3nwZgrckWg"><strong>About The Opportunity</strong></h3> <div data-page-id="JREHd8SRho8hzXxWRLOl2fqJg8v" data-lark-html-role="root" data-docx-has-block-data="false"> <div class="ace-line ace-line old-record-id-Ra5ZdxbRcoWFz4xHvYYlU4bNgJd">Building machine learning systems for risk at a global crypto exchange is fundamentally different from conventional ML engineering. 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.</div> <div class="ace-line ace-line old-record-id-NKg9dFfmco6XEKx9hSDlK3ajg0g">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.<br><br></div> <div class="ace-line ace-line old-record-id-MecOdZ17qodgM6xgrholmVDsgfh">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.</div> <div class="ace-line ace-line old-record-id-QXYHd7SRUoci7OxcAMRl2SRAgLg">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.</div> <div class="ace-line ace-line old-record-id-QXYHd7SRUoci7OxcAMRl2SRAgLg">&nbsp;</div> <div class="ace-line ace-line old-record-id-QXYHd7SRUoci7OxcAMRl2SRAgLg"> <div data-page-id="JREHd8SRho8hzXxWRLOl2fqJg8v" data-lark-html-role="root" data-docx-has-block-data="false"> <h3 class="heading-2 ace-line old-record-id-NSpcdWGxTorINKxFj9ll5g2kgtg">What You’ll Be Doing</h3> <ul class="list-bullet1"> <li class="ace-line ace-line old-record-id-QcXndk3t9o5627xNsc4lDX7mg4b" data-list="bullet"> <div>Design, build, and deploy machine learning models for risk use cases such as payment fraud, account takeover, scam detection, deposit and withdrawal risk, promotional abuse, customer risk assessment, and transaction monitoring.</div> </li> <li class="ace-line ace-line old-record-id-IGMtddGNhovFLLxtI2Bl2Dj0gGe" data-list="bullet"> <div>Own production ML systems end to end, including feature pipelines, training workflows, model serving, decision integrations, monitoring, alerting, drift detection, retraining, and incident response.</div> </li> <li class="ace-line ace-line old-record-id-SZeJdrRYUocYBpxJ9DvleuYrgPc" data-list="bullet"> <div>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.</div> </li> <li class="ace-line ace-line old-record-id-Oq0adwl5HooDq5xnvLrl7YURgSe" data-list="bullet"> <div>Work closely with risk operations teams to understand investigation workflows, incorporate reviewer feedback, improve model explainability, and continuously refine labels and training data.</div> </li> <li class="ace-line ace-line old-record-id-Rpw2dElrSoVQnVxvj8Bl1kVIgye" data-list="bullet"> <div>Apply AI-assisted development throughout the engineering workflow, using LLM coding tools to accelerate implementation, testing, debugging, analysis, and documentation while maintaining appropriate security and review standards.</div> </li> <li class="ace-line ace-line old-record-id-ZWChdW9SLo9AJvxud8qlgljsgRd" data-list="bullet"> <div>Develop AI-powered risk capabilities such as investigation agents, case summarization, evidence…
Skills asked for
- machine learning
- html
- llm
- data science
- python
- pytorch
- tensorflow
- scikit-learn
Similar jobs
- Senior / Staff Turbomachinery Systems EngineerAntora · San Jose
Your next role is already in here.
Search live openings from thousands of employers, save the ones worth a second look, and let JobBob keep watch for the rest.