agent-based modeling and the future of programming

[ineffableAlias]

i'm considering to translate notebooks from Netlogo https://github.com/NetLogo/NetLogo and python's Mesa https://github.com/projectmesa/mesa into Jax in order to approximate the cutting-edge of scientific computing and ML/DL to ABM and economics and i'm looking for Jax experts to review those translations. my plan is to do it in pure jax for generality and explicitness. thank you in advance!