NLP(初级)
JAX(Pytrees)
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)来构建和训练深度学习模型。