Using jax scan for PPO + atari + envpool XLA#328
Merged
51616 merged 17 commits intovwxyzjn:masterfrom 51616:ppo_jax_scanDec 21, 2022
+626-3
Commits
Commits on Dec 3, 2022
- committed
- committed
- committed
Commits on Dec 4, 2022
Commits on Dec 9, 2022
- committed
Commits on Dec 12, 2022
Commits on Dec 20, 2022
- committed
- committed
- committed
- committed
- authored
