摘要：和上一篇 SAC In JAX 一样，我们用 JAX 实现 VAE，配置一样，只需要安装符合版本的 torchvision 即可，实现中提供了 tensorboard 记录以及最后的可视化展示，测试集即为最经典的 MNIST，代码如下： import jax import flax import m 阅读全文

摘要：众所周知，SAC 是 RL 中的一种高效的 Off Policy 算法，在《动手学强化学习》中已经给出了比较完善的实现。而 JAX 是一种新兴的神经网络范式，以函数式编程为基础，这里将以《动手学强化学习》中的实现为范本，实现一个 SAC In JAX，同时配套 tensorboard 与 model 阅读全文