5-7 years of experience in machine learning engineering or applied data science, with a proven track record of productionizing models.
Proficient in Python and familiar with ML frameworks like PyTorch, TensorFlow, and scikit-learn.
Strong grasp of machine learning fundamentals, including supervised learning and anomaly detection.
Experience with AI-assisted engineering and LLM coding tools, balancing productivity with security risks.
Knowledge of model explainability techniques such as SHAP and feature attribution.
Hands-on experience in designing and deploying LLM agents for operational workflows.
Familiarity with integrating LLM agents with internal systems and APIs.
Responsibilities
Design, build, and deploy machine learning models for various risk use cases.
Own end-to-end production ML systems, including monitoring and incident response.
Collaborate with risk strategy teams to implement effective production controls.
Work with risk operations to enhance model explainability and refine training data.
Utilize AI-assisted development tools to streamline engineering workflows.
Develop AI-powered capabilities for risk investigation and decision support.
Ensure models are explainable and well-documented for stakeholders.
Benefits
Opportunity to work on cutting-edge AI and machine learning technologies.
Collaborative environment with cross-functional teams.
Focus on high-stakes domains like fraud and risk management.
Access to advanced tools and frameworks for AI development.
Potential for professional growth in a rapidly evolving field.
Full Job Description
What You'll Be Doing
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.
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.
Work closely with risk operations teams to understand investigation workflows, incorporate reviewer feedback, improve model explainability, and continuously refine labels and training data.
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.
Develop AI-powered risk capabilities such as investigation agents, case summarization, evidence collection, review recommendations, alert triage, suspicious-entity mining, and automated decision support.
Take research-stage models into reliable production systems by validating feature logic, reviewing data quality, addressing latency and scalability constraints, and ensuring consistency between offline training and online inference.
Ensure models and decision systems are explainable, traceable, and well documented so that model outputs can be understood by risk operations, product stakeholders, internal governance teams, and regulators where applicable.
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. Ensure LLM agents operate safely in a regulated risk environment by implementing permission controls, audit logs, data privacy protections, prompt and tool security, fallback mechanisms, and clear escalation paths.
What We Look For in You
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. Scope and level will be calibrated based on experience.
Strong Python skills and hands-on experience with machine learning frameworks such as PyTorch, TensorFlow, XGBoost, LightGBM, or scikit-learn.
Strong knowledge of applied machine learning fundamentals, including supervised learning, anomaly detection, representation learning, class-imbalanced modeling, model calibration, and evaluation under changing data distributions.
Demonstrable fluency with AI-assisted engineering. 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.
Familiarity with model explainability techniques such as SHAP, feature attribution, reason-code generation, and model scorecards.
Hands-on experience designing and deploying production LLM agents, including agentic workflows, tool calling, retrieval-augmented generation, prompt and context management, structured output generation, and multi-step task orchestration.
Experience integrating LLM agents with internal systems, APIs, databases, search tools, case-management platforms, or decision engines to automate complex operational workflows.
A strong understanding of LLM-agent evaluation and reliability, including hallucination control, grounding, observability, permissions, failure handling, human review, latency, and cost optimization.
Experience building AI agents for fraud, risk, compliance, customer operations, cybersecurity, or other high-stakes domains is a meaningful advantage.
Strong communication and collaboration skills, with the ability to work effectively with engineers, data scientists, risk specialists, product managers, operations teams, and legal or compliance stakeholders.
Notice:
All official OKX vacancies are published on this website. While roles may appear on selected third-party platforms from time to time, information on other sites may be inaccurate or outdated. If in doubt, please apply directly through our official careers website.
Information collected and processed as part of the recruitment process of any job application you choose to submit is subject to OKX's Candidate Privacy Notice.
About OKX
OKX Careers
There has never been a more opportune time to join the dynamic team at OKX—the forefront of innovation in the digital asset exchange market.
Work You’ll Do
Join OKX's market-leading team to assist some of the most sophisticated and forward-thinking traders in mastering the fast-paced world of cryptocurrency trading and investment.
Transform the landscape of digital finance with the brightest minds at OKX, where innovation meets deep industry expertise in a culture of diversity and leadership.
Lead from a unique position in the marketplace, at the nexus of technology, financial expertise, and digital innovation.
Work with a global team of business and technology advisors to help clients and the company navigate through the complexities of digital finance.
Collaborate with a dedicated group of professionals in an environment where career growth and opportunities are abundant.
Introducing the OKX Professional Growth and Opportunities
The team is building a market-leading position to help drive the future of digital finance, offering job opportunities that foster professional and personal growth.
Do Innovative Work
Join one of the most innovative teams in the financial technology sector, engaging in projects that push the boundaries of what is possible in digital transactions.
Drive Innovation and Leadership
Deliver targeted solutions on a trusted platform through a depth and breadth of experience and innovation that’s second to none.
Be Part of a Great Team
Engage in a wide range of projects utilizing cutting-edge technology and harness the unparalleled capabilities, global scale, and joint solution development of a powerhouse team.
Future-proof Your Career
Advance your career with limitless opportunities in a company that supports unmatched training, development, and certification support.
Explore
Discover how OKX is leading the way in blockchain technology and cryptocurrency exchange solutions, providing robust tools for trading, risk management, and investment.
The OKX Leadership and Innovation Edge
The combined service capabilities, global scale, and joint solution development help clients and the company overcome challenges and lead transformation in the digital finance industry.
Clients worldwide look to OKX for new strategies and solutions to drive growth and innovation in the digital era.
Stay Connected
Join the Team
Search open positions that match your skills and interests. OKX looks for passionate, curious, creative, and solution-driven team players.
SEARCH OKX JOBS
Keep Up to Date
Stay ahead with career tips, insider perspectives, and industry-leading insights you can put to use today—all from the people who work at OKX.
READ CAREERS BLOG
Job Alert Emails
Personalize your subscription to receive job alerts, latest news, and insider tips tailored to your preferences. Explore what exciting and rewarding opportunities await at OKX.
