pcgrl-jax.A JAX-accelerated implementation of the Procedural Content Generation via Reinforcement Learning (PCGRL) framework. We train RL agents to generate grid-based video game levels, optimizing for controllable functional heuristics, and study the out-of-distribution generalization (to new map shapes/sizes) of (multiple) agents with local observations.