谷歌狂喜：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推理訓練最快選擇

聯系作者

文章來源：新智元
作者微信：AI_era
作者簡介：智能+中國主平臺，致力于推動中國從互聯網+邁向智能+新紀元。重點關注人工智能、機器人等前沿領域發展，關注人機融合、人工智能和機器人對人類社會與文明進化的影響，領航中國新智能時代。