虽然在某种上来说,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》