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