Stoix: Distributed Single-Agent Reinforcement Learning End-to-End in JAX

What's Changed

• chore: refactor types by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/106
• Fix: Sebulba PPO Metrics by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/108
• feat: add MLP torso after CNN torso by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/110
• chore: add some extra alg run checks by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/111
• fix: make FlattenObservationWrapper also flatten next_obs by @JesseSilverberg in https://github.com/EdanToledo/Stoix/pull/115
• Edan toledo/issue112 by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/117
• Feat/add jax env factory by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/118
• chore: move system config files into separate folders by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/119
• Fix typos by @BioGeek in https://github.com/EdanToledo/Stoix/pull/120
• feat: add SPS for trainer by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/129
• fix: logger accepts arrays as data and use tensorboard_logger Logger by @Your-Cheese in https://github.com/EdanToledo/Stoix/pull/131
• feat: add slurm launcher by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/133
• fix: add specific system condition for envpool installation by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/124
• chore: add uuid to uniqueid by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/134
• Feat/add r2d2 by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/137
• Feat/add impala by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/138
• chore: Redundant duplicate tests and linters workflow in PR's by @mmcaulif in https://github.com/EdanToledo/Stoix/pull/149
• fix: issue with GAE calc and add test case by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/148
• feat: add readme for algorithms by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/150
• fix: change fbx api by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/151
• feat: pass config env kwargs to gymnax via envparams by @emergenz in https://github.com/EdanToledo/Stoix/pull/152
• Chore/make wrappers work sebulba by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/141
• feat: add spo by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/144
• Improve sebulba by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/161
• Fix/gae calc by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/160
• fix: do not remove hydra logging handler by @emergenz in https://github.com/EdanToledo/Stoix/pull/158
• Feat/add single torso impala by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/164
• feat: Migration to UV by @mmcaulif in https://github.com/EdanToledo/Stoix/pull/155
• Feat/pqn by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/122
• feat: add optional obs norm to ff_ppo.py by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/167
• [FIX] Fix the logger if there are no streamhandlers by @Michael-Beukman in https://github.com/EdanToledo/Stoix/pull/173
• [FEATURE] Kinetix by @Michael-Beukman in https://github.com/EdanToledo/Stoix/pull/172
• Fix the Kinetix pixels config by @Michael-Beukman in https://github.com/EdanToledo/Stoix/pull/175
• Feat/use stoa wrappers by @EdanToledo in https://github.com/EdanToledo/Stoix/pull/177

New Contributors

• @JesseSilverberg made their first contribution in https://github.com/EdanToledo/Stoix/pull/115
• @BioGeek made their first contribution in https://github.com/EdanToledo/Stoix/pull/120
• @Your-Cheese made their first contribution in https://github.com/EdanToledo/Stoix/pull/131
• @mmcaulif made their first contribution in https://github.com/EdanToledo/Stoix/pull/149
• @emergenz made their first contribution in https://github.com/EdanToledo/Stoix/pull/152
• @Michael-Beukman made their first contribution in https://github.com/EdanToledo/Stoix/pull/173

Full Changelog: https://github.com/EdanToledo/Stoix/compare/v0.0.2...v0.0.3