摘要：
JAX 中文文档（七） 原文：jax.readthedocs.io/en/latest/ 使用 shard_map 的 SPMD 多设备并行性 原文：jax.readthedocs.io/en/latest/notebooks/shard_map.html shard_map 是一种单程序多数据（S 阅读全文
摘要：
JAX 中文文档（六） 原文：jax.readthedocs.io/en/latest/ 高级教程 原文：jax.readthedocs.io/en/latest/advanced_guide.html 本节包含更高级主题的示例和教程，如多核计算、自定义操作及更深入的应用 示例 使用 tensorf 阅读全文
摘要：
JAX 中文文档（九） 原文：jax.readthedocs.io/en/latest/ 使用jax.checkpoint控制自动微分的保存数值（又名jax.remat） 原文：jax.readthedocs.io/en/latest/notebooks/autodiff_remat.html im 阅读全文
摘要：
JAX 中文文档（二） 原文：jax.readthedocs.io/en/latest/ JAX 教程 原文：jax.readthedocs.io/en/latest/tutorials.html 快速入门 关键概念 即时编译 自动向量化 自动微分 调试入门 伪随机数 使用 pytrees 工作 分 阅读全文
摘要：
JAX 中文文档（八） 原文：jax.readthedocs.io/en/latest/ 自动微分手册 原文：jax.readthedocs.io/en/latest/notebooks/autodiff_cookbook.html alexbw@, mattjj@ JAX 拥有非常通用的自动微分系 阅读全文