Senior Staff Engineer, AI Platform and 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
About the Role
Lead the design, development, and evolution of large-scale AI infrastructure for machine learning and generative AI workloads. Shape platform strategy, build systems for training and deployment, solve architecture and performance challenges, evaluate emerging technologies, and mentor senior engineers.
Requirements
- Bachelor's degree in Computer Science, Information Technology, Engineering, Mathematics, or equivalent
- At least 7+ years of relevant industry experience
- Experience in AI Agent or RAG engineering solutions
- Experience developing microservices and APIs with Java, Spring Boot, Spring Cloud, Python, or LangChain
- Knowledge of multithreading and distributed architecture
- Knowledge of messaging frameworks such as Kafka
- Knowledge of database solutions including vector databases
- Experience with AI models or multi-agentic AI systems
Responsibilities
- Architect and build large-scale AI infrastructure for training, fine-tuning, and deployment
- Define platform standards and reference architectures for distributed training, inference, and model lifecycle management
- Design scalable systems across compute, storage, networking, and orchestration layers
- Drive performance, reliability, and cost optimization for AI workloads
- Translate requirements into robust platform capabilities
- Set technical direction and best practices for AI platform development
- Evaluate and integrate emerging AI infrastructure technologies
- Mentor senior engineers and influence architecture
Benefits
- Competitive total compensation package
- L&D programs and education subsidy
- Team building programs and company events
- Wellness and meal allowances
- Comprehensive healthcare schemes for employees and dependants
