JAX MARL Environments

Experimental environments for coordination, foraging, migration, and fluid populations.

I build JAX-based MARL environments for studying coordination and population structure. These include fluid versions of predator-prey and level-based foraging, as well as environments where movement, spawning, resource availability, and group structure interact.

The goal is to make population change experimentally visible: not just how agents act, but how many agents should exist, when they should be created, and who should create them.

Keywords: JAX, MARL environments, spawning, foraging, predator-prey, population dynamics.