JAX 入门#

欢迎来到 JAX!JAX 文档包含许多有用的入门资源。快速入门 是最容易上手并了解 JAX 项目概况的地方。

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

教程#

如果您已准备好更深入地探索 JAX,JAX 教程将更详细地介绍

如果您喜欢视频介绍,这里有 JAX 贡献者 Jake VanderPlas 的一个视频

构建于 JAX 之上#

JAX 为更广泛社区开发的许多工具提供了核心数值计算原语。例如,如果您有兴趣使用 JAX 训练神经网络,那么两个受到良好支持的选项是 FlaxHaiku

有关社区策划的跨越广泛领域的 JAX 相关项目列表,请查看 Awesome JAX

查找帮助#

如果您对 JAX 有疑问,我们很乐意回答!以下是两个可以获得问题解答的好地方