Staff Machine Learning Engineer

OKX is seeking a Staff Machine Learning Engineer to build and deploy ML systems for risk management at a global crypto exchange. You will work across the full ML lifecycle, from problem formulation to real-time deployment, and help shape AI usage within the risk organization, including LLM-assisted development and AI-powered investigations.

• Design, build, and deploy ML models for various risk use cases
• Own production ML systems end-to-end
• Partner with risk strategy and product teams to translate models into effective production controls
• Apply AI-assisted development using LLM coding tools
• Develop AI-powered risk capabilities like investigation agents and automated decision support
• Build and deploy LLM-based agents for risk operations and investigation workflows