JAX: 快如 PyTorch,简单如 NumPy - 深度学习与数据科学

JAX: 快如 PyTorch,简单如 NumPy - 深度学习与数据科学

JAX 是 TensorFlow 和 PyTorch 的新竞争对手。 JAX 强调简单性而不牺牲速度和可扩展性。由于 JAX 需要更少的样板代码,因此程序更短、更接近数学,因此更容易理解。 长话短说: 使用 import jax.numpy 访问 NumPy 函数,使用 import jax.sci...

谷歌开源计算框架JAX:比Numpy快30倍,还可在TPU上运行!

谷歌开源计算框架JAX:比Numpy快30倍,还可在TPU上运行!

 相信大家对numpy, Tensorflow, Pytorch已经极其熟悉,不过,你知道JAX吗? JAX发布之后,有网友进行了测试,发现,使用JAX,Numpy运算可以快三十多倍! 下面是使用Numpy的运行情况:import numpy as np # 使用标准n...

Python 科学计算库 NumPy 快速入门

14 课时 |
40592 人已学 |
免费
开发者课程背景图

本页面内关键词为智能算法引擎基于机器学习所生成,如有任何问题,可在页面下方点击"联系我们"与我们沟通。

社区圈子

大数据
大数据
大数据计算实践乐园,近距离学习前沿技术
188891+人已加入
加入