扫码查看

谷歌做的数值计算库,自动微分配合XLA编译,搞科研写新算子的人特别爱用。

JAX
微信微博邮箱复制链接