Sacha Chernyavskiy

Usage example ```python import jax from pogema import GridConfig, pogema_v0 grid_config = GridConfig( size=8, num_agents=5, obs_radius=2, seed=9, on_target="finish", max_episode_steps=128, integration="jax", ) env = pogema_v0(grid_config=grid_config) key = jax.random.key(0) # resetting state,...