备注

本节包含有关使用 JAX 的主题的简短说明；另请参阅JAX 增强提案 (JEP)中的更长的设计讨论。

依赖项和版本兼容性

• API 兼容性概述了 JAX 在各个版本之间的 API 兼容性方面的政策。

• Python 和 NumPy 版本支持政策概述了 JAX 在与 Python 和 NumPy 兼容性方面的政策。

迁移和弃用

• jax.Array 迁移总结了 jax v 0.4.1 中默认数组类型的更改。

内存和计算使用

• 异步调度描述了 JAX 的异步调度模型。

• 并发描述了 JAX 如何与其他 Python 并发交互。

• GPU 内存分配描述了 JAX 如何与 GPU 上的内存分配交互。

程序员护栏

• 秩提升警告描述了如何配置jax.numpy以避免隐式秩提升。