Multi-agent simulator in Jax for research and teaching in AI & ALife - View it on GitHub