Skip to main content

Atari: CleanRL PPO + JAX + EnvPool's XLA vs openai/baselins' PPO (part 1)

Created on September 10|Last edited on September 10

2M4M6M8MSteps0100020003000400050006000Episodic Return
20406080100120Time (minutes)0100020003000400050006000Episodic Return
CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
5



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
5



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3



CleanRL ppo_atari_envpool_xla_jax.py
3
openai/baselines' PPO
3