Jax 和训练神经网络

问题描述

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

解决方法

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