说明

说明#

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

依赖项和版本兼容性
迁移和弃用
内存和计算使用
  • 异步调度 描述了 JAX 的异步调度模型。

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

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

程序员护栏