标签: dm-haiku

亚麻和dm-haiku之间的主要区别是什么？

从他们的描述来看：

• Flax，JAX 的神经网络库
• Haiku，受 Sonnet 启发的 JAX 神经网络库

问题：

我应该选择哪一个基于 jax 的库来实现，比如说DeepSpeech模型（由 CNN 层 + LSTM 层 + FC 组成）和 ctc-loss？

UPD。

找到dm-haiku的开发者关于差异的解释：

Flax 包含更多的电池，并配有优化器、混合精度和一些训练循环（我听说这些是解耦的，你可以根据需要使用尽可能多或尽可能少的量）。Haiku 的目标只是解决 NN 模块和状态管理，它将问题的其他部分留给其他库（例如用于优化的 optax）。

Haiku 被设计为 Sonnet（一个 TF NN 库）到 JAX 的端口。因此，如果（像 DeepMind 一样）您有大量可能想要在 JAX 中使用的 Sonnet+TF 代码，并且您希望尽可能轻松地（在任一方向上）迁移该代码，Haiku 是一个更好的选择。

我认为否则这取决于个人喜好。在 Alphabet 中，每个库都有数百名研究人员使用，所以我认为无论哪种方式都不会出错。在 DeepMind，我们对俳句进行了标准化，因为它对我们来说有意义。我建议查看两个库提供的示例代码，看看哪个符合您构建实验的偏好。我想如果您将来改变主意，您会发现将代码从一个库移动到另一个库并不是很复杂。

原来的问题仍然相关。