Sacha Chernyavskiy
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,...
Whenever I try to install the library with `pip` or from source (having no other dependencies installed, plain docker container), the process is not successful because it can't resolve requirements...
Hey! It's not the main PR, so it might be deleted later but to summarise: 1. Replaced `Sonnet's` initialisation with native `Pytorch`'s. Implementations with jax and torch had different structures,...
Hello @lanctot ! We @parvxh, me, and @harmanagrawal present our first parts of the AlphaZero refactor. We (with the major help of the guys) have rewritten the models, using `flax.linen`...
Whom it may concern, Do you think that: 1) adding [`uv`](https://docs.astral.sh/uv/) in addition to pip to support the integration with the modern dependency systems 2) with addition of `jax` versions...