Python 和 NumPy 版本支持策略#

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

对于 Python 版本支持,我们从用户那里了解到,36 个月的支持窗口可能太短,例如,由于新的 CPython 版本传播到 Linux 供应商版本存在延迟。因此,JAX 支持 Python 版本的时间至少比 SPEC-0 建议的时间长九个月。

这意味着我们至少支持

  • 在每个 JAX 版本之前的 45 个月内的所有 Python 功能版本。例如

    • Python 3.10 于 2021 年 10 月发布,将在新的 JAX 版本中至少支持到 2025 年 7 月

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

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

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

  • 在每个 JAX 版本之前的 24 个月内的所有 NumPy 功能版本。例如

    • NumPy 1.25 于 2023 年 6 月发布,将在新的 JAX 版本中至少支持到 2025 年 6 月

    • NumPy 1.26 于 2023 年 9 月发布,将在新的 JAX 版本中至少支持到 2025 年 9 月

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

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

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

  • 在每个 JAX 版本之前的 24 个月内的所有 SciPy 功能版本。例如

    • Scipy 1.11 于 2023 年 6 月发布,将在新的 JAX 版本中至少支持到 2025 年 6 月

    • Scipy 1.12 于 2024 年 1 月发布,将在新的 JAX 版本中至少支持到 2026 年 1 月

    • Scipy 1.13 于 2024 年 4 月发布,将在新的 JAX 版本中至少支持到 2026 年 4 月

    • Scipy 1.14 于 2024 年 6 月发布,将在新的 JAX 版本中至少支持到 2026 年 6 月

    • Scipy 1.15 于 2025 年 1 月发布,将在新的 JAX 版本中至少支持到 2027 年 1 月

JAX 版本可能支持比此策略严格要求的更旧版本的 Python、NumPy 和 SciPy,但超出列出的日期后,可能会随时停止对旧版本的支持。