Commit Graph

3 Commits

Author SHA1 Message Date
Yash Katariya da0f2de14d Replace jnp.DeviceArray with the new public type jax.Array.
PiperOrigin-RevId: 477558995
2022-10-12 10:32:06 +00:00
Jake VanderPlas 6fcb84268e Use jax.tree_util.tree_map in place of deprecated tree_multimap.
The latter is a simple alias of the former, so this change is a no-op.

PiperOrigin-RevId: 461229165
2022-07-24 17:53:28 +01:00
Georg Ostrovski 5f6b11299e Added tandem_dqn to deepmind-research repo
PiperOrigin-RevId: 405653277
2021-10-26 16:30:00 +01:00