Skip to main content

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

Created on September 10|Last edited on September 10

2M4M6M8MSteps01000020000300004000050000Episodic Return
20406080100120140Time (minutes)0100002000030000400005000060000Episodic 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
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