我们是否需要另一个 JAX 神经网络库？

大家好，今天我想和大家聊聊我一直在开发的一个新的 JAX 库，但在开始之前，我想先讨论这个话题：我们是否需要另一个 JAX 神经网络库？

JAX 库

JAX 神经网络库种类繁多，从像 Flax 和 Haiku 这样的函数式库，到像 Equinox 这样基于 Pytree 的库，应有尽有。