vwxyzjn / cleanrl

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
http://docs.cleanrl.dev
Other
5.02k stars 575 forks source link

Benchmark `dqn_jax.py` using CPU only #317

Closed vwxyzjn closed 1 year ago

vwxyzjn commented 1 year ago

Problem Description

See https://github.com/vwxyzjn/cleanrl/pull/315#discussion_r1020974503