主要区别在于flax和dm-haiku的描述: - Flax是一个基于JAX的神经网络库。 - Haiku是一个受Sonnet启发的基于JAX的神经网络库。 问题: 如果我要实现一个由CNN层+LSTM层+FC组成并带有CTC损失的DeepSpeech模型,应该选择哪个基于JAX的库呢? 更新. 从dm-haiku的开发者那里找到了关于区别的解释: Flax更加全面,内置了优化...