摘要：
JAX 中文文档（十三） 原文：jax.readthedocs.io/en/latest/ 在 JAX 之上构建 原文：jax.readthedocs.io/en/latest/building_on_jax.html 学习高级 JAX 使用的一种很好的方法是看看其他库如何使用 JAX，它们如何将库 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(337)
评论(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
绝不原创的飞龙
阅读(81)
评论(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
绝不原创的飞龙
阅读(327)
评论(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
绝不原创的飞龙
阅读(169)
评论(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
绝不原创的飞龙
阅读(221)
评论(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
绝不原创的飞龙
阅读(289)
评论(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
绝不原创的飞龙
阅读(148)
评论(0)
推荐(0)
摘要：
JAX 中文文档（六） 原文：jax.readthedocs.io/en/latest/ 高级教程 原文：jax.readthedocs.io/en/latest/advanced_guide.html 本节包含更高级主题的示例和教程，如多核计算、自定义操作及更深入的应用 示例 使用 tensorf 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(191)
评论(0)
推荐(0)
摘要：
JAX 中文文档（九） 原文：jax.readthedocs.io/en/latest/ 使用jax.checkpoint控制自动微分的保存数值（又名jax.remat） 原文：jax.readthedocs.io/en/latest/notebooks/autodiff_remat.html im 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(122)
评论(0)
推荐(0)
摘要：
JAX 中文文档（二） 原文：jax.readthedocs.io/en/latest/ JAX 教程 原文：jax.readthedocs.io/en/latest/tutorials.html 快速入门 关键概念 即时编译 自动向量化 自动微分 调试入门 伪随机数 使用 pytrees 工作 分 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(375)
评论(0)
推荐(0)
浙公网安备 33010602011771号