简单来说 ， 可以这样理解：当你想处理一些对算力要求很高的问题时 ， 通过JAX ， 你可以将复杂的问题快速分散到多个TPU上 。
目前 ， 谷歌大脑的Trax、Flax、Jax-md ， 以及DeepMind的神经网络库Haiku和强化学习库RLax等 ， 都是基于JAX构建的 。
值得一提的是 ， JAX诞生之际 ， 正是PyTorch在学术界强烈冲击TensorFlow之时 。
加州大学伯克利分校旗下研究机构RISELab的数据显示：
2019年1月到6月底 ， 在arXiv上发表的论文中 ， 提及TensorFlow和PyTorch的数量相差无几 ， PyTorch仅稍稍落后 。
不过在增长速度方面 ， 与2018年1-6月相比 ， PyTorch的“份额”增长了194% 。 TensorFlow则只增长了23% 。

文章图片
参考链接：
https://blog.tensorflow.org/2022/06/%20bringing-machine-learning-to-every-developers-toolbox.html?m=1

• b站官方小视频app今日停运，网络崩溃了
• amd线程撕裂者pro5000wx系列官方定价公布
• AMD3月份发布了基于Zen3架构的新一代锐龙线程撕裂者PRO5000WX系列|amd线程撕裂者pro5000wx官方定价，价格离谱
• 今天上午|红魔7s系列游戏手机官方发文：7月见面
• 15年做糊21款硬件，谷歌到底栽在哪儿？
• AMD|AMD Zen3线程撕裂者官方价格公布：64核心贵的离谱！
• 中国电信|女儿为已故父亲按时交话费被停机引热议 对该行为不满：电信回应难服众
• |某科技公司称大专生只值200元？招聘平台回应来了：不许说这话
• 小米|首发1英寸大底IMX989！小米12S系列官方样张出炉：夜景无敌
• realme|官方暗示：真我大师探索版或将由LV设计师打造