cleanrl
cleanrl copied to clipboard
JAX + DQN
See #218. @yooceii and @kinalmehta have expressed interest in working on this. @kinalmehta is also interested in C51 (#221).
@kinalmehta if working with DQN first helps with working on C51, feel free to also work on this issue (saying this because usually working with DQN helps with "warm-up"). That way we will have two DQN PR from both you and @yooceii, and we can review and reconcile the differences between the two PRs.