摘要：
JAX 中文文档（十） 原文：jax.readthedocs.io/en/latest/ JAX 中的广义卷积 原文：jax.readthedocs.io/en/latest/notebooks/convolutions.html JAX 提供了多种接口来跨数据计算卷积，包括： jax.numpy. 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(227)
评论(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
绝不原创的飞龙
阅读(299)
评论(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
绝不原创的飞龙
阅读(161)
评论(0)
推荐(0)
摘要：
JAX 中文文档（六） 原文：jax.readthedocs.io/en/latest/ 高级教程 原文：jax.readthedocs.io/en/latest/advanced_guide.html 本节包含更高级主题的示例和教程，如多核计算、自定义操作及更深入的应用 示例 使用 tensorf 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(194)
评论(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
绝不原创的飞龙
阅读(127)
评论(0)
推荐(0)
摘要：
JAX 中文文档（二） 原文：jax.readthedocs.io/en/latest/ JAX 教程 原文：jax.readthedocs.io/en/latest/tutorials.html 快速入门 关键概念 即时编译 自动向量化 自动微分 调试入门 伪随机数 使用 pytrees 工作 分 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(382)
评论(0)
推荐(0)
摘要：
JAX 中文文档（八） 原文：jax.readthedocs.io/en/latest/ 自动微分手册 原文：jax.readthedocs.io/en/latest/notebooks/autodiff_cookbook.html alexbw@, mattjj@ JAX 拥有非常通用的自动微分系 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(224)
评论(0)
推荐(0)
浙公网安备 33010602011771号