Python 和 NumPy 版本支持策略

对于 NumPy 和 SciPy 版本支持，JAX 遵循 Python 科学计算社区的 SPEC 0 规范。

关于 Python 版本支持，我们从用户处了解到，36 个月的支持窗口可能太短，例如由于新的 CPython 版本发布到 Linux 发行版存在延迟。因此，JAX 对 Python 版本的支持时长比 SPEC-0 建议的时间至少多出九个月。

这意味着我们至少支持：

• 每次 JAX 发布前 45 个月内的所有 Python 功能版本。例如：

  • Python 3.11 于 2022 年 10 月发布，在新版 JAX 中将至少支持到 2026 年 7 月。

  • Python 3.12 于 2023 年 10 月发布，在新版 JAX 中将至少支持到 2027 年 7 月。

  • Python 3.13 于 2024 年 10 月发布，在新版 JAX 中将至少支持到 2028 年 7 月。

  • Python 3.14 于 2025 年 10 月发布，在新版 JAX 中将至少支持到 2029 年 7 月。

• 每次 JAX 发布前 24 个月内的所有 NumPy 功能版本。例如：

  • NumPy 2.0 于 2024 年 6 月发布，在新版 JAX 中将至少支持到 2026 年 6 月。

  • NumPy 2.1 于 2024 年 8 月发布，在新版 JAX 中将至少支持到 2026 年 8 月。

  • NumPy 2.2 于 2024 年 12 月发布，在新版 JAX 中将至少支持到 2026 年 12 月。

  • NumPy 2.3 于 2025 年 6 月发布，在新版 JAX 中将至少支持到 2027 年 6 月。

  • NumPy 2.4 于 2025 年 12 月发布，在新版 JAX 中将至少支持到 2027 年 12 月。

• 每次 JAX 发布前 24 个月内的所有 SciPy 功能版本。例如：

  • SciPy 1.14 于 2024 年 6 月发布，在新版 JAX 中将至少支持到 2026 年 6 月。

  • SciPy 1.15 于 2025 年 1 月发布，在新版 JAX 中将至少支持到 2027 年 1 月。

  • SciPy 1.16 于 2025 年 6 月发布，在新版 JAX 中将至少支持到 2027 年 6 月。

  • SciPy 1.17 于 2026 年 1 月发布，在新版 JAX 中将至少支持到 2028 年 1 月

JAX 发布版本可能支持比此政策严格要求更旧的 Python、NumPy 和 SciPy 版本，但在上述日期之后，对旧版本的支持可能会随时终止。