AI Native Platform Architect

About the role

This role focuses on leading the architectural design and development of an AI Agent platform at Okx. The architect will be responsible for integrating AI Agents and large language models into the company's systems, building the knowledge augmentation layer, and establishing the runtime and governance framework for these AI agents. The work involves driving AI Native capabilities within engineering workflows.

Success in this position means delivering an enterprise-grade AI platform that supports Okx's objectives. The architect will collaborate closely with product, engineering, QA, and operations teams to ensure the platform meets various requirements and integrates smoothly into existing infrastructure. This involves applying a strong understanding of system design and AI architectures.

The ideal candidate will have over eight years of experience in backend, platform, or infrastructure engineering, with proficiency in Java/Python. A strong grasp of LLM and AI Agent architectures, along with practical experience with tools like LangChain and LangGraph, is essential. Understanding DevOps, CI/CD, Kubernetes, and cloud-native technologies will also be important for this role.

The listed salary range for this position is between $180,000 and $218,900 USD.

Skills that matter here

• Java: Proficiency in Java is required for system design and development within the AI platform.
• Python: Proficiency in Python is required for system design and development within the AI platform.
• LangChain: Hands-on experience with LangChain is necessary for integrating large language models and AI agents.
• LangGraph: Hands-on experience with LangGraph is necessary for integrating large language models and AI agents.
• Kubernetes: A good understanding of Kubernetes is needed for managing the deployment and scaling of the AI platform.
• cloud-native: A good understanding of cloud-native technologies is required for designing and implementing the AI platform infrastructure.

Who this role suits

• A person who thrives on leading complex architectural projects from conception to delivery.
• Someone who enjoys collaborating across multiple technical and product teams.
• An individual with a deep interest in AI Agent architectures and large language models.
• A professional who values establishing robust governance and runtime frameworks for new technologies.

From the employer

• Lead the architecture design and development of the AI Agent platform.
• Drive AI Native capabilities in engineering workflows.
• Build the knowledge augmentation layer for the platform.
• Establish the runtime and governance framework for AI Agents.
• Collaborate with product, engineering, QA, and operations teams.

• Bachelor's degree or above in Computer Science, Software Engineering, AI, or related fields.
• 8+ years of experience in backend, platform, or infrastructure engineering.
• Proficient in Java/Python with strong system design skills.
• Strong understanding of LLM and AI Agent architectures.
• Hands-on experience with LangChain and LangGraph.
• Experience in designing RAG/GraphRAG/knowledge graph solutions.
• Good understanding of DevOps, CI/CD, Kubernetes, and cloud-native technologies.

• 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.