JAX:Python+NumPy 程序的可组合转换-开源

上传者: 42108054 | 上传时间: 2021-08-07 14:06:54 | 文件大小: 9.75MB | 文件类型: ZIP
借助 Autograd 的更新版本,JAX 可以自动区分原生 Python 和 NumPy 函数。 它可以通过循环、分支、递归和闭包进行微分,并且可以取导数的导数的导数。 它支持通过 grad 进行反向模式微分(又名反向传播)以及前向模式微分,并且两者可以任意组合成任何顺序。 新的是 JAX 使用 XLA 在 GPU 和 TPU 上编译和运行您的 NumPy 程序。 默认情况下,编译发生在幕后,库调用会及时编译和执行。 但是 JAX 还允许您使用单函数 API jit 将自己的 Python 函数及时编译到 XLA 优化的内核中。 编译和自动微分可以任意组合,所以你可以在不离开 Python 的情况下表达复杂的算法并获得最大的性能。

文件下载

资源详情

[{"title":"( 490 个子文件 9.75MB ) JAX:Python+NumPy 程序的可组合转换-开源","children":[{"title":"setup.py <span style='color:#111;'> 2.49KB </span>","children":null,"spread":false},{"title":"pylintrc <span style='color:#111;'> 1.63KB </span>","children":null,"spread":false},{"title":".gitignore <span style='color:#111;'> 219B </span>","children":null,"spread":false},{"title":"conftest.py <span style='color:#111;'> 852B </span>","children":null,"spread":false},{"title":"jax_logo_250px.png <span style='color:#111;'> 33.23KB </span>","children":null,"spread":false},{"title":"......","children":null,"spread":false},{"title":"<span style='color:steelblue;'>文件过多,未全部展示</span>","children":null,"spread":false}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明