JAX v0.11.1
JAX v0.11.1 发布，新增对过期导出反序列化的错误检查，并添加配置标志 jax_export_deserialize_expired_versions 以临时绕过。新增 jax.numpy.top_k，对应 NumPy v2.6.0 的 numpy.top_k。移除 exec_time_optimization_effort 和 memory_fitting_effort 标志，改用 EffortLevel 枚举。不再支持反序列化 2026 年 1 月 15 日之前导出的模块，因为该日期起仅支持 NamedSharding 序列化。jnp.take_along_axis 现在默认 wrap_negative_indices 为 True。
发展脉络
- 首次出现JAX v0.11.1JAX
- 行业反馈v0.11.1LangSmith SDK
- 行业反馈typescript-v0.9.11DeepEval
- 当前判断JAX 作为 AI 框架，其版本更新反映了对稳定性和兼容性的重视。通过收紧导出兼容性窗口，JAX 推动用户采用最新格式，减少长期维护负担。这体现了框架在快速迭代中平衡创新与稳定性的策略。Agent Pulse · 分析
JAX v0.11.1 发布，包含新功能、破坏性变更和弃用。新增对过期导出反序列化的错误检查，并添加配置标志以临时绕过。新增 jax.numpy.top_k，对应 NumPy v2.6.0。破坏性变更包括移除 exec_time_optimization_effort 和 memory_fitting_effort 标志，改用 EffortLevel 枚举；不再支持反序列化 2026 年 1 月 15 日之前的导出，因为该日期起仅支持 NamedSharding 序列化；jnp.take_along_axis 默认 wrap_negative_indices 为 True。这些变更影响依赖旧导出格式或标志的用户。
JAX 强化了导出兼容性边界，通过错误检查防止过期导出导致的下游错误，并提供了临时绕过标志。移除旧标志并引入 EffortLevel 枚举，简化了优化和内存配置。仅支持 NamedSharding 序列化，统一了分片表示。这些变更要求开发者更新导出流程和配置代码。
JAX 作为 AI 框架，其版本更新反映了对稳定性和兼容性的重视。通过收紧导出兼容性窗口，JAX 推动用户采用最新格式，减少长期维护负担。这体现了框架在快速迭代中平衡创新与稳定性的策略。
对于依赖 JAX 的团队，此更新可能带来迁移成本，但长期看提升系统稳定性。采用最新版本可减少因过期导出导致的故障，降低维护成本。
未来 JAX 可能进一步收紧兼容性窗口，并推动更多 API 标准化。开发者应关注 EffortLevel 枚举的扩展和 NamedSharding 的普及，及时更新依赖。