JAX 教程草稿# 注意 下面的教程仍在进行中；目前，请参考较旧的教程内容，包括JAX 入门，用户指南，以及现已删除的JAX 101 教程。 JAX 101# 大部分已在教程中完成！ 关键概念 即时编译 自动向量化 自动微分 调试简介 伪随机数 使用 Pytree 并行编程简介 有状态计算 示例：编写一个简单的神经网络 JAX 201# 并行计算 高级调试 性能分析和性能 JAX 301# 高级编译
