谷歌狂喜：JAX性能超越Pytorch、TensorFlow！或成GPU推理训练最快选择

AIGC动态欢迎阅读

原标题：谷歌狂喜：JAX性能超越Pytorch、TensorFlow！或成GPU推理训练最快选择
关键字：模型,性能,基准,测试,版本
文章来源：新智元
内容字数：4250字

内容摘要：

新智元报道编辑：编辑部
【新智元导读】JAX在最近的基准测试中的性能已经不声不响地超过了Pytorch和TensorFlow，也许未来会有更多的大模型诞生在这个平台上。谷歌在背后的默默付出终于得到了回报。谷歌力推的JAX在最近的基准测试中性能已经超过Pytorch和TensorFlow，7项指标排名第一。
而且测试并不是在JAX性能表现最好的TPU上完成的。
虽然现在在开发者中，Pytorch依然比Tensorflow更受欢迎。
但未来，也许有更多的大模型会基于JAX平台进行训练和运行。
模型最近，Keras团队为三个后端（TensorFlow、JAX、PyTorch）与原生PyTorch实现以及搭配TensorFlow的Keras 2进行了基准测试。
首先，他们为生成式和非生成式人工智能任务选择了一组主流的计算机视觉和自然语言处理模型：
对于模型的Keras版本，其采用了KerasCV和KerasNLP中已有的实现进行构建。而对于原生的PyTorch版本，则选择了网络上最流行的几个选项：
– 来自HuggingFace Transformers的BERT、Gemma、Mistral

原文链接：谷歌狂喜：JAX性能超越Pytorch、TensorFlow！或成GPU推理训练最快选择