GoogleJAX是一个用于变换数值函数的机器学习框架，Google称其为为结合了修改版本的Autograd（通过函数微分自动获得梯度函数）和TensorFlow的XLA（加速线性代数）。该框架的设计尽可能遵循NumPy的结构和工作流程，并与TensorFlow和PyTorch等各种现有框架协同工作。
JAX的主要功能是包括：
- grad：自动微分
- jit：编译
- vmap：自动矢量化
- pmap：SPMD编程
©️版权声明：本站展示所有网站均来自56分类目录网络转载或由站长自行提供收录，其原创性以及文中陈述文字和内容未经本网站证实，对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本网站不作任何保证或承诺，不承担任何保证责任，请读者仅作参考，并请自行核实相关内容。如果以上内容侵犯您的版权或者非授权发布和其它问题请联系本站删除，谢谢。
类似于JAX的工具
56分类目录收录了上万个全网知名网站，并每日不断更新和添加最新优秀网站，帮助你更快的找到你想要的网站，高效完成浏览！
Ctrl + D 或 ⌘ + D 收藏本站到浏览器书签栏。
