Migrate example Acme agents from TF to JAX
kenjitoyama opened this issue · comments
We currently use TF agents in our Acme example. JAX has received more attention lately so should probably encourage researchers to use that instead.
google-deepmind / android_env
RL research on Android devices.
https://www.deepmind.com/publications/androidenv-the-android-learning-environment