Yay!

Happy to share that @cleanrl_lib now supports Random Network Distillation envpool, it's 3× faster than our first version without envpool and still have comparable performance to the original implementation, say 👋 to the long training time on hard-exploration games! Details👇

Thanks to @_joaogui1's awesome contribution 🙏, @cleanrl_lib now has a TD3 JAX implementation that is 2-4x faster than the TD3 @PyTorch equivalent 🔥. Running on TPU is now possible, too 🚀! 📜 docs: docs.cleanrl.dev/rl-algorith… 💾 code: github.com/vwxyzjn/cleanrl/b… A short 🧵1/x