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 版本,但对旧版本的支持可能在列出日期之后随时停止。