摘要:
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
绝不原创的飞龙
阅读(162)
评论(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
绝不原创的飞龙
阅读(211)
评论(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
绝不原创的飞龙
阅读(280)
评论(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
绝不原创的飞龙
阅读(145)
评论(0)
推荐(0)
摘要:
JAX 中文文档(六) 原文:jax.readthedocs.io/en/latest/ 高级教程 原文:jax.readthedocs.io/en/latest/advanced_guide.html 本节包含更高级主题的示例和教程,如多核计算、自定义操作及更深入的应用 示例 使用 tensorf 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(184)
评论(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
绝不原创的飞龙
阅读(116)
评论(0)
推荐(0)
摘要:
JAX 中文文档(二) 原文:jax.readthedocs.io/en/latest/ JAX 教程 原文:jax.readthedocs.io/en/latest/tutorials.html 快速入门 关键概念 即时编译 自动向量化 自动微分 调试入门 伪随机数 使用 pytrees 工作 分 阅读全文
posted @ 2024-06-21 14:07
绝不原创的飞龙
阅读(366)
评论(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
绝不原创的飞龙
阅读(199)
评论(0)
推荐(0)

浙公网安备 33010602011771号