输入一个热词，快速查看术语含义、常见场景和相关概念。
集中展示常见热词解释，方便按概念和场景继续浏览。
JAX是谷歌开发的一个用于高性能数值计算和机器学习的Python库。它结合了NumPy熟悉的API、函数式编程思想、自动微分（Autograd）以及XLA编译器，能够将Python和NumPy代码高效地编译并运行在CPU、GPU和TPU上。其核心目标是让研究人员能够以灵活、可组合的方式，轻松编写复杂模型并实现极致的计算性能。
近期常被查询的 AI 概念。
