JAX(Autodiff)
今天,我们将研究另一个重要概念自动微分。我们已经在TensorFlow
中看到了自动微分。自动微分的想法在所有框架中都非常相似,但IMO JAX
比所有框架都做得更好。
JAX(VMAP & PMAP)
如果我了解TensorFlow/Torch
,为什么还要去学习JAX
?尽管有n
个理由,但我将介绍JAX
中的一个概念,足以说服您尝试一下。我们将研究自动矢量化。我们接下来将讨论两种转换,即vmap
和pmap
。
JAX(JIT)
JAX(PRNG)
JAX(Pure Functions)
JAX(DeviceArray)
JAX
是一个高性能机器学习库。JAX
在加速器(例如GPU
和TPU
)上编译并运行NumPy
代码。您可以使用JAX
(以及为JAX
构建的神经网络库FLAX
)来构建和训练深度学习模型。
TensorFlow & JAX
JAX
是一个高性能机器学习库。JAX
在加速器(例如GPU
和TPU
)上编译并运行NumPy
代码。您可以使用JAX
(以及为JAX
构建的神经网络库FLAX
)来构建和训练深度学习模型。
游戏AI & 强化学习
游戏AI
环境设置
游戏环境配备了已经为您实现的代理。要查看这些默认代理的列表,请运行:
1 | from kaggle_environments import make, evaluate |