IDEAS home Printed from https://ideas.repec.org/p/arx/papers/2308.13289.html
   My bibliography  Save this paper

JAX-LOB: A GPU-Accelerated limit order book simulator to unlock large scale reinforcement learning for trading

Author

Listed:
  • Sascha Frey
  • Kang Li
  • Peer Nagy
  • Silvia Sapora
  • Chris Lu
  • Stefan Zohren
  • Jakob Foerster
  • Anisoara Calinescu

Abstract

Financial exchanges across the world use limit order books (LOBs) to process orders and match trades. For research purposes it is important to have large scale efficient simulators of LOB dynamics. LOB simulators have previously been implemented in the context of agent-based models (ABMs), reinforcement learning (RL) environments, and generative models, processing order flows from historical data sets and hand-crafted agents alike. For many applications, there is a requirement for processing multiple books, either for the calibration of ABMs or for the training of RL agents. We showcase the first GPU-enabled LOB simulator designed to process thousands of books in parallel, with a notably reduced per-message processing time. The implementation of our simulator - JAX-LOB - is based on design choices that aim to best exploit the powers of JAX without compromising on the realism of LOB-related mechanisms. We integrate JAX-LOB with other JAX packages, to provide an example of how one may address an optimal execution problem with reinforcement learning, and to share some preliminary results from end-to-end RL training on GPUs.

Suggested Citation

  • Sascha Frey & Kang Li & Peer Nagy & Silvia Sapora & Chris Lu & Stefan Zohren & Jakob Foerster & Anisoara Calinescu, 2023. "JAX-LOB: A GPU-Accelerated limit order book simulator to unlock large scale reinforcement learning for trading," Papers 2308.13289, arXiv.org.
  • Handle: RePEc:arx:papers:2308.13289
    as

    Download full text from publisher

    File URL: http://arxiv.org/pdf/2308.13289
    File Function: Latest version
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Peter Belcak & Jan-Peter Calliess & Stefan Zohren, 2020. "Fast Agent-Based Simulation Framework with Applications to Reinforcement Learning and the Study of Trading Latency Effects," Papers 2008.07871, arXiv.org, revised Sep 2022.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Zhenglong Li & Vincent Tam & Kwan L. Yeung, 2024. "Developing A Multi-Agent and Self-Adaptive Framework with Deep Reinforcement Learning for Dynamic Portfolio Risk Management," Papers 2402.00515, arXiv.org, revised Sep 2024.
    2. Zijian Shi & John Cartlidge, 2021. "The Limit Order Book Recreation Model (LOBRM): An Extended Analysis," Papers 2107.00534, arXiv.org.
    3. Samuel N. Cohen & Derek Snow & Lukasz Szpruch, 2021. "Black-box model risk in finance," Papers 2102.04757, arXiv.org.
    4. Zihao Zhang & Bryan Lim & Stefan Zohren, 2021. "Deep Learning for Market by Order Data," Papers 2102.08811, arXiv.org, revised Jul 2021.

    More about this item

    NEP fields

    This paper has been announced in the following NEP Reports:

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:arx:papers:2308.13289. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: arXiv administrators (email available below). General contact details of provider: http://arxiv.org/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.