摘要：
JAX 中文文档（四） 原文：jax.readthedocs.io/en/latest/ 理解 Jaxpr 原文：jax.readthedocs.io/en/latest/jaxpr.html 更新日期：2020 年 5 月 3 日（提交标识为 f1a46fe）。 从概念上讲，可以将 JAX 转换看 阅读全文
摘要：
JAX 中文文档（十一） 原文：jax.readthedocs.io/en/latest/ JAX 增强提案（JEPs） 原文：jax.readthedocs.io/en/latest/jep/index.html 大多数改动可以通过简单的问题/讨论和拉取请求进行讨论。 然而，有些变更范围较大或需要 阅读全文
摘要：
JAX 中文文档（十五） 原文：jax.readthedocs.io/en/latest/ jax.tree 模块 原文：jax.readthedocs.io/en/latest/jax.tree.html 用于处理树形容器数据结构的实用工具。 jax.tree 命名空间包含了来自 jax.tree 阅读全文
摘要：
JAX 中文文档（十四） 原文：jax.readthedocs.io/en/latest/ jax.scipy 模块 原文：jax.readthedocs.io/en/latest/jax.scipy.html jax.scipy.cluster | vq(obs, code_book[, chec 阅读全文
摘要：
JAX 中文文档（十三） 原文：jax.readthedocs.io/en/latest/ 在 JAX 之上构建 原文：jax.readthedocs.io/en/latest/building_on_jax.html 学习高级 JAX 使用的一种很好的方法是看看其他库如何使用 JAX，它们如何将库 阅读全文
摘要：
JAX 中文文档（十七） 原文：jax.readthedocs.io/en/latest/ JAX 术语表 原文：jax.readthedocs.io/en/latest/glossary.html 数组 JAX 的 numpy.ndarray 的类比。见 jax.Array。 CPU 缩写Cent 阅读全文
摘要：
JAX 中文文档（十六） 原文：jax.readthedocs.io/en/latest/ jax.experimental.sparse.bcoo_multiply_dense 原文：jax.readthedocs.io/en/latest/_autosummary/jax.experimenta 阅读全文
摘要：
JAX 中文文档（十二） 原文：jax.readthedocs.io/en/latest/ Jax 和 Jaxlib 版本控制 原文：jax.readthedocs.io/en/latest/jep/9419-jax-versioning.html 为什么 jax 和 jaxlib 是独立的包？ 我 阅读全文
摘要：
JAX 中文文档（十） 原文：jax.readthedocs.io/en/latest/ JAX 中的广义卷积 原文：jax.readthedocs.io/en/latest/notebooks/convolutions.html JAX 提供了多种接口来跨数据计算卷积，包括： jax.numpy. 阅读全文
摘要：
JAX 中文文档（三） 原文：jax.readthedocs.io/en/latest/ 有状态计算 原文：jax.readthedocs.io/en/latest/stateful-computations.html JAX 的转换（如jit()、vmap()、grad()）要求它们包装的函数是纯 阅读全文