评论: 0 浏览: 1659 ***新更新时间: 2个月前 (06-12)
查看
GoogleJAX是一个用于变换数值函数的机器学习框架，Google称其为为结合了修改版本的Autograd（通过函数微分自动获得梯度函数）和TensorFlow的XLA（加速线性代数）。该框架的设计尽可能遵循NumPy的结构和工作流程，并与TensorFlow和PyTorch等各种现有框架协同工作。
JAX的主要功能是包括：
grad：自动微分
jit：编译
vmap：自动矢量化
pmap：SPMD编程
相关阅读：
1、为什么设计pdf?
2、为什么喜欢室内设计?
3、【数学建模】基于空间模型设计的创意平板折叠桌
4、一次搞定！中级软件设计师备考通关秘籍
5、04-Spring Data中的设计模式
发表评论:取消回复
名称(*)
邮箱
网址
正文(*)
◎欢迎参与讨论，请在这里发表您的看法、交流您的观点。
发表评论:
◎欢迎参与讨论，请在这里发表您的看法、交流您的观点。