JAX

推荐指数这方面，

★★★★☆

说到最佳使用场景。

GoogleJAX是一个用于变换数值函数的机器学习框架，Google称其为为结合了修改版本的Autograd和TensorFlow的XLA。该框架的设计尽可能遵循NumPy的结构和工作流程，并与TensorFlow和PyTorch等各种现有框架协同工作。话说回来，

主要功能的观点是，

JAX的主要功能是包括：

• grad的观点是。自动微分
• jit这方面，编译
• 从vmap来看，自动矢量化
• pmap这方面，SPMD编程

。