Senior/Staff Engineer, AI Agent Development

OKX

Full Time Singapore

AI Job Summary

Design and deliver core architecture for an enterprise AI Agent platform integrating large language models, knowledge systems, and software development workflows. You'll lead implementation of AI capabilities in coding, solution generation, and test automation while establishing governance frameworks for agent execution and cost control. Requires 6+ years backend or platform engineering experience, proficiency in Java or Python, strong LLM and AI Agent knowledge, hands-on experience with LangChain and LangGraph, and RAG or knowledge graph expertise. Based in Singapore with right to work.

Written from this posting by Neural Jobs AI. The full description is below.

AI Resume Tailoring Sign in to use this AI AI Cover Letter Sign in to use this AI

Job Description

OKX will be prioritising applicants who have a current right to work in Singapore, and do not require OKX's sponsorship of a visa.

Who We Are

At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom.

OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps). OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets. We are safe and reliable, backed by our Proof of Reserves.

Across our multiple offices globally, we are united by our core principles: We Before Me, Do the Right Thing, and Get Things Done. These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er. OKX is part of OKG, a group that brings the value of Blockchain to users around the world, through our leading products OKX, OKX Wallet, OKLink and more.

About the Opportunity

You will be responsible for the core architecture design and technical delivery of an enterprise-grade AI Native engineering platform, driving the deep integration of large language models, AI Agents, knowledge capabilities, and the software development lifecycle. The platform will support intelligent capabilities across requirement analysis, solution generation, coding collaboration, test automation, and environment orchestration.This role focuses on engineering-grade AI implementation in complex R&D scenarios. You are expected to turn model capabilities into reusable, scalable, and governable platform capabilities, and continuously drive the evolution of the engineering platform toward an AI Native architecture.

What You’ll Be Doing

Lead the overall architecture design and core capability development of the AI Agent platform, including task orchestration, tool invocation, context management, memory mechanisms, and multi-agent collaboration.
Drive the implementation of AI Native capabilities in engineering workflows, including AI coding, technical solution generation, test case generation, test report generation, and automated environment deployment.
Build the knowledge augmentation layer for the platform, including the design, integration, and application of RAG, GraphRAG, and knowledge graph capabilities.
Establish the runtime and governance framework for AI Agents, including execution isolation, permission control, auditability, evaluation, stability, and cost governance.
Collaborate with product, engineering, QA, and operations teams to standardize, productionize, and scale platform capabilities, continuously improving engineering efficiency and delivery quality.

What We Look For In You

Bachelor’s degree or above, preferably in Computer Science, Software Engineering, Artificial Intelligence, or related fields.
6+ years of experience in backend engineering, platform engineering, or infrastructure engineering, with strong system design and delivery capabilities.
Proficient in Java / Python, with solid coding skills, architectural thinking, and experience building complex systems.
Strong understanding of LLM and AI Agent architectures, including task decomposition, tool use, context management, memory, execution loops, and exception handling.
Hands-on experience with LangChain and LangGraph, and strong familiarity with mainstream AI coding tools or frameworks such as Claude Code, Codex, Cursor, and Copilot.
Experience in designing or developing RAG / GraphRAG / knowledge graph solutions for enterprise knowledge augmentation and engineering use cases.
Good understanding of DevOps, CI/CD, Kubernetes, and cloud-native technologies, with strong cross-functional collaboration and execution skills.

Nice to Haves

Experience contributing to open-source projects related to AI Agents, AI Coding, RAG, or knowledge graphs.
Experience building enterprise engineering platforms, DevOps platforms, or AI platforms.
Familiarity with open-source Agent projects such as openclaw, openhands/openharness, or Hermes.

Perks & Benefits

Competitive total compensation package
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!

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.

Do you match this job?

Check my match

OKX

Financial Services & Fintech · United States

At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom. OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps). OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets. We are safe and reliable, backed by our Proof of Reserves.

All jobs at OKX

Location

Singapore

Job Overview

Job Posted:

4 months ago

Workplace

On-site

Job Type

Full Time

Education

Any

Experience

8+ years

Senior/Staff Engineer, AI Agent Development

AI Job Summary

Job Description

Who We Are

About the Opportunity

What You’ll Be Doing

What We Look For In You

Nice to Haves

Perks & Benefits

Location

Company

Talent

Employer

Support

Senior/Staff Engineer, AI Agent Development

AI Job Summary

Job Description

Who We Are

About the Opportunity

What You’ll Be Doing

What We Look For In You

Nice to Haves

Perks & Benefits

Location

Share This Job:

Related Jobs

Company

Talent

Employer

Support