注释

本节包含关于使用 JAX 的相关主题的简短笔记；另请参阅 JAX 增强提案 (JEPs) 中的更长设计讨论。

依赖关系和版本兼容性：

• API 兼容性 概述了 JAX 在不同版本之间关于 API 兼容性的政策。

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

迁移和弃用：

• jax.Array 迁移 总结了 jax v 0.4.1 中默认数组类型的变化

内存和计算使用：

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

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

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

程序员防护栏：

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