Principal Engineer, Agent Infrastructure & Memory Architecture at OKX

<h3 class="heading-3 ace-line old-record-id-LgPLdbHBpoytXax9JbZlBfsZgCY"><strong>Who We Are</strong></h3> <div class="ace-line ace-line old-record-id-XaCddqdpToiZjgxDzTflWQnngk5">At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom.<br><br>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.<br><br>Across our multiple offices globally, we are united by our core principles:&nbsp;<em>We Before Me</em>,&nbsp;<em>Do the Right Thing</em>, and&nbsp;<em>Get Things Done</em>. These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er.</div> <div class="ace-line ace-line old-record-id-XaCddqdpToiZjgxDzTflWQnngk5"> <h3 class="heading-2 ace-line old-record-id-doxusl417D129Ked7hp9WBVv9lg"><strong>About the Opportunity</strong></h3> <div data-page-id="YnvfdojQpornTcxhh7FlZ37ugze" data-lark-html-role="root" data-docx-has-block-data="false"> <div class="ace-line ace-line old-record-id-Foo4dvQATotKkuxbUEglUO6sg2J"> <p data-start="170" data-end="482">We are seeking an&nbsp;<strong>AI Infrastructure Engineer</strong>&nbsp;to design and build the foundational systems that power next-generation AI agents. This role sits at the intersection of research and production engineering, focusing on scalable&nbsp;<strong>long-term memory</strong>,&nbsp;<strong>knowledge graph infrastructure</strong>, and robust&nbsp;<strong>context management systems</strong>.</p> <p data-start="484" data-end="704">You will transform cutting-edge research concepts (e.g., Context Graphs, GraphRAG, agent memory architectures) into auditable, maintainable, and production-grade infrastructure that enables reliable, stateful AI systems.</p> </div> </div> <h3 class="heading-2 ace-line old-record-id-doxusI2RkaIWtuQsGjyJjA6hybg"><strong>What You’ll Be Doing&nbsp;</strong></h3> <div data-page-id="YnvfdojQpornTcxhh7FlZ37ugze" data-lark-html-role="root" data-docx-has-block-data="false"> <ul class="list-bullet1"> <li class="ace-line ace-line old-record-id-JLKOdU6cwoyIjbxLKZxlEttugxc" data-list="bullet"> <div>Architect and implement long-term memory infrastructure for AI agents, including auditable, versioned, and rollback-capable middleware that supports persistent, stateful reasoning.</div> </li> <li class="ace-line ace-line old-record-id-AFvSdoB44oXpewxukzelPublgCe" data-list="bullet">Design and own end-to-end GraphRAG and knowledge graph pipelines, from ingestion and schema design to graph indexing, retrieval optimization, and integration with LLM-based reasoning systems.</li> <li class="ace-line ace-line old-record-id-JXHrdNTcUoZBvWxvj1HlVENig3d" data-list="bullet">Build scalable context management infrastructure, elevating context from prompt-level techniques to structured, maintainable systems such as Context Graphs that enable long-horizon and cross-session reasoning.</li> <li class="ace-line ace-line old-record-id-WnzhdfL28oLSntxM2KilxZrBgCg" data-list="bullet">Translate advanced agent research into production-grade systems, prototyping emerging architectures and establishing best practices for memory, retrieval, and infrastructure reliability.</li> <li class="ace-line ace-line old-record-id-PKU2dkSMhoPMEuxuUiclXKKAgmg" data-list="bullet">Ensure reliability, observability, and governance of agent infrastructure, including traceable memory operations, evaluation frameworks, and safeguards against drift and degradation.</li> </ul> </div> <h3 class="heading-2 ace-line old-record-id-doxusA9TF9jUkqvydDVZ26era1g"><strong>What We Look For In You&nbsp;</strong></h3> </div> <ul class="list-bullet1"> <li class="ace-line ace-line old-record-id-KJNudcXf2oSMFRxU5dUl5tcxgsf" data-list="bullet">PhD or Master's in Computer Science, AI, Machine Learning, Robotics or related field with at least&nbsp;<strong>10 years of industry experience</strong>.</li> <li class="ace-line ace-line old-record-id-KJNudcXf2oSMFRxU5dUl5tcxgsf" data-list="bullet">Possess a deep expertise in at least one of the following areas: <strong>long-term memory</strong>,&nbsp;<strong>knowledge graphs</strong>, or&nbsp;<strong>context management</strong>.</li> </ul> <div class="ace-line ace-line old-record-id-XaCddqdpToiZjgxDzTflWQnngk5"> <h3 class="heading-2 ace-line old-record-id-doxusvnFLHEcrU3EZSqi1qdAlRh"><strong>Nice to Haves</strong></h3> <div data-page-id="YnvfdojQpornTcxhh7FlZ37ugze" data-lark-html-role="root" data-docx-has-block-data="false"> <ul class="list-bullet1"> <li class="ace-line ace-line old-record-id-VoBZdvOhNomtYkxWILjl8wUVgQ1" data-list="bullet"> <div>Experience in top-tier AI labs or Tech industry leaders</div> </li> <li class="ace-line ace-line old-record-id-Q4K1dT1Ynof7ttxDIHVln7Wjg1c" data-list="bullet"> <div>Experience in high-frequency trading, fintech, Web3, or crypto platforms is a strong advantage</div> </li> <li class="ace-line ace-line old-record-id-OIrUdcJaCo83dfx32OflI47Sgnc" data-list="bullet"> <div>Startups or research-intensive environments preferred; must be comfortable operating with ambiguity and speed</div> </li> </ul> </div> </div> <div class="ace-line ace-line old-record-id-XaCddqdpToiZjgxDzTflWQnngk5"> <h3 class="heading-3 ace-line old-record-id-Xfu6dMaUkoTGqGx6arYu3QhKsZe"><strong>OKX Statement</strong></h3> <div class="ace-line ace-line old-record-id-BDVHdvrnCoy11xx8RoVuyc2As3g">The salary range for this position is&nbsp;<span class="text-only text-font-italic" data-eleid="22">$313,055.00 to $450,000.00</span>. The salary offered depends on a variety of factors, including job-related knowledge, skills, experience, and market location. In addition to the salary, a performance bonus and long-term incentives may be provided as part of the compensation package, as well as a full range of medical, financial, and/or other benefits, dependent on the position offered. Applicants should apply via Okcoin and OKX internal or external careers site.</div> <div class="ace-line ace-line old-record-id-ZfHydXbNIo2qmdxmQCru8ZDksBd">&nbsp;</div> <div class="ace-line ace-line old-record-id-BWwndhjsUozjW6xZoNxuILk6s9e">OKX is committed to equal employment opportunities regardless of race, color, genetic information, creed, religion, sex, sexual orientation, gender identity, lawful alien status, national origin, age, marital status, and non-job related physical or mental disability, or protected veteran status. Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.</div> <div class="ace-line ace-line old-record-id-BWwndhjsUozjW6xZoNxuILk6s9e">&nbsp;</div> <div class="ace-line ace-line old-record-id-BWwndhjsUozjW6xZoNxuILk6s9e">&nbsp;</div> </div><div class="content-conclusion"><div data-lark-html-role="root"><span class="text-only" data-eleid="18"><span class="text-only"><span class="text-only" data-eleid="6">Notice:<br></span></span></span> <div data-lark-html-role="root"><span class="text-only" data-eleid="26"><span class="text-only">All official </span><span class="text-only text-with-abbreviation text-with-abbreviation-bottomline">OKX</span><span class="text-only"> vacancies are published on this website.</span></span> <span class="text-only" data-eleid="28"><span class="text-only">While roles may appear on selected third-party platforms from time to time, information on other sites may be inaccurate or outdated. </span></span><strong><span class="text-only" data-eleid="29"><span class="text-only">If in doubt, please apply directly through our official careers website.</span></span></strong></div> </div> <div data-lark-html-role="root"><span class="text-only" data-eleid="18"><span class="text-only">Information collected and processed as part of the recruitment process of any job application you choose to submit is subject to&nbsp;</span><span class="text-only text-with-abbreviation text-with-abbreviation-bottomline">OKX</span><span class="text-only">'s </span></span><a class="link rich-text-anchor __anchor-intercept-flag__ text-content-link" href="https://www.okx.com/en-eu/help/okx-candidate-privacy-notice" target="_blank" data-eleid="19" data-lark-is-custom="true" data-lark-link="true">Candidate Privacy Notice</a><span class="text-only" data-eleid="20"><span class="text-only">.</span></span></div></div>