说明#
本节包含有关使用 JAX 的主题的简短说明;另请参阅 JAX 增强提案 (JEP) 中更长的设计讨论。
- 依赖项和版本兼容性
API 兼容性 概述了 JAX 在不同版本之间 API 兼容性的策略。
Python 和 NumPy 版本支持策略 概述了 JAX 在与 Python 和 NumPy 的兼容性方面的策略。
- 迁移和弃用
jax.Array 迁移 总结了 jax v 0.4.1 中默认数组类型的更改
- 内存和计算使用
- 程序员护栏
本节包含有关使用 JAX 的主题的简短说明;另请参阅 JAX 增强提案 (JEP) 中更长的设计讨论。
API 兼容性 概述了 JAX 在不同版本之间 API 兼容性的策略。
Python 和 NumPy 版本支持策略 概述了 JAX 在与 Python 和 NumPy 的兼容性方面的策略。
jax.Array 迁移 总结了 jax v 0.4.1 中默认数组类型的更改