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