CAX (Cellular Automata Accelerated in JAX)는 다양한 과학 분야에서 등장과 자기 조직화 연구에 기여하는 셀룰러 오토마타(CA) 연구를 가속화하기 위해 설계된 고성능, 유연한 오픈소스 라이브러리입니다. 하드웨어 가속을 통해 최첨단 성능을 제공하며, 모듈식 아키텍처, 직관적인 API, 그리고 임의 차원의 이산 및 연속 CA 지원을 통해 유연성을 유지합니다. 기본적인 셀룰러 오토마타 모델부터 신경 셀룰러 오토마타 및 MNIST 숫자 자체 분류와 같은 고급 애플리케이션까지 다양한 벤치마크와 애플리케이션을 통해 CAX의 성능과 유연성을 보여줍니다. 시뮬레이션 속도를 최대 2000배까지 향상시키며, 간단한 1차원 CA가 1D-ARC 챌린지에서 GPT-4를 능가하는 것을 보여줍니다.