This is a fork of JaxMARL with minimal changes: Changed Python dependency to 3.9 Changed package dependency versions: jax, jaxlib