Google在放弃Tensorflow后推出Google版本的Pytorch框架 —— Jax

相关：

谷歌放弃TensorFlow了

虽然在某种上来说，Google已经放弃了Tensorflow框架，但是由于Tensorflow的历史地位所造成的遗留应用，因此Tensorflow框架已经处于维护和更新阶段，但是这也都是为历史用户而服务的，可以说Tensorflow的用户数量一直处于下降状态而不是上升状态，所以这也所另一方面印证了Google放弃Tensorflow的必要性。

但是，Google虽然放弃了Tensorflow但是没有放弃深度学习框架领域，因而Google推出了自研的类似pytorch的Jax框架。准确的来说Jax是pytorch的无神经网络模型API的形式，因此与其说是深度学习框架更应该说是科学计算框架。

pytorch是怎么来的呢，其实是现有的numpy，然后有了NVIDIA版本的numpy，即cupy，然后在cupy基础上构建了chainer框架，然后在此基础上构建出了pytorch，而Google推出的Jax，其功能与cupy更为贴近，当然Google在推出jax的同时也推出了基于其上的神经网络框架，因此在这个意义上来说Google在将pytorch特性融入到TensorFlow失败后便全新的推出了一款自研版本的类pytorch的Jax框架。

---

Jax的主页：

https://jax.readthedocs.io/en/latest/index.html

关于Jax的论文，见：

《Compilation of machine learning software through high-level traceability》