Jax 和训练神经网络

Question

我是 JAX 的初学者，我正在尝试学习如何训练神经网络。我看了一些博客，但据我所知，没有一个库可以轻松地训练它，比如 sklearn 中的 'fit'。我对分类任务很感兴趣，能否请您向我推荐任何博客以便在我的问题中采用 his/her 算法？

Answer 1

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