python - Jax 和训练神经网络

Question

我是 JAX 的初学者，我正在尝试学习如何训练神经网络。我看过一些博客，但据我所知，没有一个可以轻松训练它的库，比如 sklearn 中的“fit”。我对分类任务感兴趣，你能推荐我任何博客，以便将他/她的算法应用到我的问题中吗？

Answer 1

JAX 是一个数组操作库，而不是一个深度学习库：在这方面，您应该认为它更类似于 NumPy，而不是类似于 scikit-learn。如果您想要基于 JAX 构建的神经网络，还有其他几个不错的项目可用，例如haiku和flax。