March 12, 2025
Cellular Automata Accelerated in JAX
[Submitted on 3 Oct 2024 (v1), last revised 11 Mar 2025 (this version, v2)] View a PDF of the paper titled CAX: Cellular Automata Accelerated in JAX, by Maxence Faldor and 1 other authors View PDF HTML (experimental) Abstract:Cellular automata have become a cornerstone for investigating emergence and self-organization across diverse scientific disciplines. However, the absence of a hardware-accelerated cellular automata library limits the exploration of new research directions, hinders collaboration, and impedes reproducibility. In this work, we introduce CAX (Cellular Automata Accelerated in JAX), a high-performance and flexible open-source library designed to accelerate cellular automata research. CAX delivers cutting-edge