开始使用 JAX

欢迎来到 JAX！JAX 文档包含了许多有用的资源，帮助你入门。快速入门 是最容易入门并了解 JAX 项目概览的地方。

如果你习惯于编写 NumPy 代码并开始探索 JAX，你可能会发现以下资源很有帮助：

• 如何在 JAX 中思考 是对 JAX 执行模型的概念性介绍。

• 🔪 JAX - 锐利的部分 🔪 列出了 JAX 的一些尖锐角落。

• 常见问题解答 (FAQ) 回答了一些常见的 jax 问题。

教程

如果你准备更深入地探索 JAX，JAX 教程会提供更多详细信息：

• 教程
  • 快速入门
  • 关键概念
  • 即时编译
  • 自动向量化
  • 自动微分
  • 调试简介
  • 伪随机数
  • 使用 pytrees
  • 并行编程简介
  • 有状态计算

如果你更喜欢视频介绍，这里有一段来自JAX贡献者Jake VanderPlas的视频：

基于 JAX 构建

JAX 提供了为大型社区开发的多种工具提供核心数值计算原语。例如，如果您对使用 JAX 进行神经网络训练感兴趣，两个得到良好支持的选项是 Flax 和 Haiku。

对于一个由社区策划的跨多个领域的JAX相关项目列表，请查看 Awesome JAX。

寻求帮助

如果你对 JAX 有疑问，我们很乐意解答！有两个不错的问答地点是：

• JAX GitHub 讨论

• JAX on StackOverflow