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.24 於 2022 年 12 月發布,並將在新 JAX 版本中支援至 2024 年 12 月

    • 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 月

  • 每個 JAX 版本發布前 24 個月內的所有 SciPy 功能版本。例如

    • Scipy 1.10 於 2023 年 1 月發布,並將在新 JAX 版本中支援至 2025 年 1 月

    • 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 月

JAX 版本可能支援比此政策嚴格要求的更舊版本的 Python、NumPy 和 SciPy,但超過列出的日期後,隨時可能停止支援舊版本。