Skip to main content
costa-huang
Projects
cleanRL
Reports
Jax DQN speed regression

Jax DQN speed regression

Costa
Created on July 28|Last edited on July 28

50100150Time (seconds)100200300400500
100k200k300k400kglobal_step200025003000350040004500
dqn (torch)
1
dqn_jax (6942c97 w/ jit action sampling)
1
dqn_jax_old (89dcbb4 no jit action sampling)
1

yooceii_dqn_atari_jax is taken directly from https://wandb.ai/yooceii/dqn-atari-jax/runs/ueohm3pu

dqn_atari (torch)
1
dqn_atari_jax (6942c97 w/ jit action sampling)
1
dqn_atari_jax_old (89dcbb4 no jit action sampling)
1
openrlbenchmark
1
yooceii_dqn_atari_jax
1



Created with ❤️ on Weights & Biases.

https://wandb.ai/costa-huang/cleanRL/reports/Jax-DQN-speed-regression--VmlldzoyMzkwMTQy
Made with Weights & Biases. Sign up or log in to create reports like this one.