Schelling segregation model workshop with Numba and JAX implementations - View it on GitHub