OKX - Senior Staff Engineer, AI Platform and Infrastructure
Upload My Resume
Drop here or click to browse · Tap to choose · PDF, DOCX, DOC, RTF, TXT
Requirements
• Experience in AI Agent / RAG engineering solutions • Experience in Microservices / API development with Java/SpringBoot/SpringCloud/Python/Langchain • Knowledgeable in multi-threading and distributed architecture • Understanding of mainstream messaging frameworks including Kafka etc. • Knowledge of different databases solutions including vector DB • Good to Have • Good to Have • Experience with AI models or multi-agentic AI systems • Experience in high-frequency trading, fintech, Web3, or crypto platforms is a strong advantage
Responsibilities
• Architect and build large-scale AI infrastructure supporting training, fine-tuning, and deployment of AI models • Define platform standards and reference architectures for distributed training, inference, and model lifecycle management • Lead the design of scalable systems across compute, storage, networking, and orchestration layers (e.g. GPU/accelerator clusters, schedulers, data pipelines) • Drive performance, reliability, and cost optimization for AI workloads at scale • Partner with the AI Science team, and relevant business units to translate requirements into robust platform capabilities • Set technical direction and best practices for AI platform development, including scalability, reliability, observability, security, and operational excellence • Evaluate and integrate emerging technologies in AI infrastructure (frameworks, accelerators, serving stacks, tooling) • Mentor senior engineers and act as a subject-matter-expert, influencing architecture beyond your immediate scope • What We Look For In You • Bachelors Degree in Computer Science, Information Technology, Engineering, Mathematics or equivalent, with at least 7+ years of relevant experience in the industry
Benefits
• L&D programs and Education subsidy for employees' growth and development • Various team building programs and company events • Wellness and meal allowances • Comprehensive healthcare schemes for employees and dependants • More that we love to tell you along the process! • 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. • 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.
No credit card. Takes 10 seconds.