GoogleJAX是一个用于变换数值函数的机器学习框架，Google称其为为结合了修改版本的Autograd（通过函数微分自动获得梯度函数）和TensorFlow的XLA（加速线性代数）。该框架的设计尽可能遵循NumPy的结构和工作流程，并与TensorFlow和PyTorch等各种现有框架协同工作。
JAX的主要功能是包括：
- grad：自动微分
- jit：编译
- vmap：自动矢量化
- pmap：SPMD编程
©️版权声明：
1、本站所有收录网站信息为原网站提供，版权归原网站所有；
2、本周所有收录网站如有失效或网址更新，可联系网站管理员处理；
3、本周仅提供对收录网站之信息展示平台，不承担任何与之相关法律责任；
4、本站不接受违法信息，如有违法内容，请联系管理员 立即举报；
5、商业合作或纠错，请联系QQ：993884858，有且仅有此联系方式。
1、本站所有收录网站信息为原网站提供，版权归原网站所有；
2、本周所有收录网站如有失效或网址更新，可联系网站管理员处理；
3、本周仅提供对收录网站之信息展示平台，不承担任何与之相关法律责任；
4、本站不接受违法信息，如有违法内容，请联系管理员 立即举报；
5、商业合作或纠错，请联系QQ：993884858，有且仅有此联系方式。
类似网站
四汇网提供黄金外汇门户导航,外汇网站网址查询,外汇知识学院,财经数据,外汇牌价,外汇分析软件,外汇培训,外汇交易,专家外汇评论,交易策略,外汇理财,外汇知识、动态等！
Ctrl + D 或 ⌘ + D 收藏本站到浏览器书签栏。
粤公网安备36040302000382号