JAX:高性能数组计算

JAX 是AutogradXLA,汇集在一起​​用于高性能数值计算。

熟悉的API

JAX 提供了一个熟悉的 NumPy 风格的 API,以方便研究人员和工程师采用。

转换

JAX 包括用于编译、批处理、自动微分和并行化的可组合函数转换。

到处跑

相同的代码在多个后端执行,包括 CPU、GPU 和 TPU

JAX

 

0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧