摘要：
JAX 中文文档（十一） 原文：jax.readthedocs.io/en/latest/ JAX 增强提案（JEPs） 原文：jax.readthedocs.io/en/latest/jep/index.html 大多数改动可以通过简单的问题/讨论和拉取请求进行讨论。 然而，有些变更范围较大或需要 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(78)
评论(0)
推荐(0)
摘要：
JAX 中文文档（十五） 原文：jax.readthedocs.io/en/latest/ jax.tree 模块 原文：jax.readthedocs.io/en/latest/jax.tree.html 用于处理树形容器数据结构的实用工具。 jax.tree 命名空间包含了来自 jax.tree 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(105)
评论(0)
推荐(0)
摘要：
JAX 中文文档（十四） 原文：jax.readthedocs.io/en/latest/ jax.scipy 模块 原文：jax.readthedocs.io/en/latest/jax.scipy.html jax.scipy.cluster | vq(obs, code_book[, chec 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(88)
评论(0)
推荐(0)
摘要：
JAX 中文文档（十三） 原文：jax.readthedocs.io/en/latest/ 在 JAX 之上构建 原文：jax.readthedocs.io/en/latest/building_on_jax.html 学习高级 JAX 使用的一种很好的方法是看看其他库如何使用 JAX，它们如何将库 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(271)
评论(0)
推荐(0)
摘要：
JAX 中文文档（十七） 原文：jax.readthedocs.io/en/latest/ JAX 术语表 原文：jax.readthedocs.io/en/latest/glossary.html 数组 JAX 的 numpy.ndarray 的类比。见 jax.Array。 CPU 缩写Cent 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(54)
评论(0)
推荐(0)
摘要：
JAX 中文文档（十六） 原文：jax.readthedocs.io/en/latest/ jax.experimental.sparse.bcoo_multiply_dense 原文：jax.readthedocs.io/en/latest/_autosummary/jax.experimenta 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(271)
评论(0)
推荐(0)
摘要：
JAX 中文文档（十二） 原文：jax.readthedocs.io/en/latest/ Jax 和 Jaxlib 版本控制 原文：jax.readthedocs.io/en/latest/jep/9419-jax-versioning.html 为什么 jax 和 jaxlib 是独立的包？ 我 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(134)
评论(0)
推荐(0)
摘要：
JAX 中文文档（十） 原文：jax.readthedocs.io/en/latest/ JAX 中的广义卷积 原文：jax.readthedocs.io/en/latest/notebooks/convolutions.html JAX 提供了多种接口来跨数据计算卷积，包括： jax.numpy. 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(154)
评论(0)
推荐(0)
摘要：
JAX 中文文档（三） 原文：jax.readthedocs.io/en/latest/ 有状态计算 原文：jax.readthedocs.io/en/latest/stateful-computations.html JAX 的转换（如jit()、vmap()、grad()）要求它们包装的函数是纯 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(259)
评论(0)
推荐(0)
摘要：
JAX 中文文档（七） 原文：jax.readthedocs.io/en/latest/ 使用 shard_map 的 SPMD 多设备并行性 原文：jax.readthedocs.io/en/latest/notebooks/shard_map.html shard_map 是一种单程序多数据（S 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(113)
评论(0)
推荐(0)
浙公网安备 33010602011771号