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