Foragax: An Agent-Based Modelling Framework Based on JAX. - Siddharth Chaturvedi; Ahmed ElGazzar; Marcel van Gerven (CoRR, 2024)