细胞自动机已成为跨越神经科学、人工生命和理论物理等多个科学学科研究涌现和自组织现象的基石。然而，缺乏硬件加速的细胞自动机库限制了对新研究方向的探索、阻碍了合作，并阻碍了可重复性。在这项工作中，我们介绍了 CAX（在 JAX 中加速的细胞自动机），这是一个旨在加速细胞自动机研究的高性能、灵活的开源库。CAX 通过用户友好的界面提供尖端的性能和模块化设计，并且可以支持任何维度的离散和连续细胞自动机。我们通过广泛的基准测试和应用展示了 CAX 的性能和灵活性。从经典模型（如基本细胞自动机和康威的生命游戏）到高级应用（如生长神经细胞自动机和自分类 MNIST 数字），CAX 将模拟速度提高了 2,000 倍。此外，我们展示了 CAX 加速研究的潜力，并展示了一组三个新颖的细胞自动机实验，这些实验得益于库的模块化架构，每个实验只需几行代码即可实现。值得注意的是，我们表明，一个简单的单维细胞自动机在 1D-ARC 挑战中可以胜过 GPT-4。