开源项目
JAX：v0.9.0统一并行计算API语义
jax-ml/jax
概述
JAX是jax-ml维护的开源框架，采用Apache-2.0许可，仓库提供安装指引。本次事件为v0.9.0正式版：统一pmap的输入秩语义，并让export支持显式分片，实质改变并行计算API的语义与导出机制；当期v0.8.3等修复版本不计入本次主版本事件。使用pmap与导出功能的开发者需核对新语义下现有代码的行为，迁移时按官方文档调整调用方式。
jax-ml/jax
JAX是jax-ml维护的开源框架，采用Apache-2.0许可，仓库提供安装指引。本次事件为v0.9.0正式版：统一pmap的输入秩语义，并让export支持显式分片，实质改变并行计算API的语义与导出机制；当期v0.8.3等修复版本不计入本次主版本事件。使用pmap与导出功能的开发者需核对新语义下现有代码的行为，迁移时按官方文档调整调用方式。