JAX

AI开发框架

Google推出的用于变换数值函数的机器学习框架

官方网址：https://jax.readthedocs.io/en/latest/

链接直达手机查看

广告位招租

GoogleJAX是一个用于变换数值函数的机器学习框架，Google称其为为结合了修改版本的Autograd（通过函数微分自动获得梯度函数）和TensorFlow的XLA（加速线性代数）。该框架的设计尽可能遵循NumPy的结构和工作流程，并与TensorFlow和PyTorch等各种现有框架协同工作。

JAX的主要功能是包括：

grad：自动微分
jit：编译
vmap：自动矢量化
pmap：SPMD编程

提示：若发现您的权益受到侵害，请立即联系客服，我们会尽快为您处理

类似网站

Label Studio

免费开源的数据标注工具

DL4J

开源的使用JVM部署和训练深度学习模型的套件

TensorFlow

Google推出的机器学习和人工智能开源库

JAX

Google推出的用于变换数值函数的机器学习框架

Scikit-learn

Python机器学习库

Lightning AI

快速训练、部署和开发人工智能产品的深度学习框架，由Pytorch Lightning团队推出

NLTK

Python自然语言处理工具包

Keras

Python版本的TensorFlow深度学习API

LangChain

开发由语言模型驱动的应用程序的框架

JAX