JAX

AI开发框架

JAX

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

标签： AI开发框架

访问官网

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

JAX的主要功能是包括：

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

©️版权声明：若无特殊声明，本站所有文章版权均归陆客资源网-AI工具集原创和所有，未经许可，任何个人、媒体、网站、团体不得转载、抄袭或以其他方式复制发表本站内容，或在非我站所属的服务器上建立镜像。否则，我站将依法保留追究相关法律责任的权利。

类似网站

http://www.lookae.cn/uploadfile/202312/dd59d44e68c1a2b.png

昇思MindSpore

华为开源自研AI深度学习框架

http://www.lookae.cn/uploadfile/202312/1358bdea59a1593.png

Apache MXNet

免费开源的深度学习框架

http://www.lookae.cn/uploadfile/202312/f1e32a3583ebdff.png

开源机器学习框架

http://www.lookae.cn/uploadfile/202312/016272997c45478.png

Python科学计算必备的包

http://www.lookae.cn/uploadfile/202312/11c17a8268e12a4.png

将AI快速集成到你自己的应用中

http://www.lookae.cn/uploadfile/202312/303645c30afe267.png

飞桨PaddlePaddle

开源深度学习平台

http://www.lookae.cn/uploadfile/202312/030ab801e6d7af6.png

Vercel AI SDK

Vercel开源的搭建AI聊天机器人的开发套件，支持React/Svelte/Vue等框架

http://www.lookae.cn/uploadfile/202312/9f693e46ba7912d.png

UC伯克利研究推出的深度学习框架

http://www.lookae.cn/uploadfile/202312/1e5de8718679598.png

Python版本的TensorFlow深度学习API