今日 DeepMind 研究人员宣布将在 GitHub 发布另外两个 JAX 库：用于优化的 Optax 和用于编写更好的测试和可靠代码的 Chex。Optax 旨在通过提供以自定义方式轻松重组的构建基块来促进研究。Chex 是一个实用程序库，用于帮助编写可靠的 JAX 代码。