Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
Training code and pre-trained models for Neural Cellular Automata on different datasets and tasks
