Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
Simulation of Systems of interacting mean-field Particles with High Efficiency