首頁 科技週邊 人工智慧 Keras 3.0正式發布:可用於TensorFlow、JAX和PyTorch

Keras 3.0正式發布:可用於TensorFlow、JAX和PyTorch

Nov 29, 2023 pm 03:43 PM
產業 keras 3

經過 5 個月的更新迭代,Keras 3.0 終於來了。

「大新聞:我們剛剛發布了Keras 3.0 版本!」Keras 之父François Chollet 在X 上激動的表示。 「現在你可以在JAX、TensorFlow 以及PyTorch 框架上運行Keras…」

Keras 3.0正式发布:可用于TensorFlow、JAX和PyTorch

對於這一更新,Keras 官方表示,這個版本足足花了他們5個月的時間進行公測才完成。 Keras 3.0 是對 Keras 的完全重寫,你可以在 JAX、TensorFlow 或 PyTorch 之上運行 Keras 工作流程,新版本還具有全新的大模型訓練和部署功能。你可以選擇最適合自己的框架,也可以根據當前的目標從一種框架切換到另一種框架都沒有問題。

Keras 3.0正式发布:可用于TensorFlow、JAX和PyTorch

Keras 位址:https://keras.io/keras_3/

被250 多萬開發者使用的Keras,迎來3.0 版本

#Keras API 可用於JAX、TensorFlow 和PyTorch。現有的僅使用內建層的 tf.keras 模型可以在 JAX 和 PyTorch 中運行!

Keras 3.0正式发布:可用于TensorFlow、JAX和PyTorch

Keras 3 可與任何 JAX、TensorFlow 和 PyTorch 工作流程無縫協作。 Keras 3 不僅適用於以Keras 為中心的工作流程,例如定義Keras 模型、優化器、損失和度量,它還旨在與JAX、TensorFlow 和PyTorch 低階後端本地工作流程無縫集成,在訓練Keras 模型時,你可以選擇使用JAX 訓練、TensorFlow 訓練、PyTorch 訓練,也可以將其作為JAX 或PyTorch 模型的一部分,上述操作都沒有問題。 Keras 3 在 JAX 和 PyTorch 中提供了與 tf.keras 在 TensorFlow 中相同程度的低階實作彈性。

Keras 3.0正式发布:可用于TensorFlow、JAX和PyTorch

預訓練模型。你現在可以在 Keras 3 中使用各種預訓練模型。現在已經有40 個Keras 應用模型可在後端中使用,此外,KerasCV 和KerasNLP 中存在的大量預訓練模型(例如BERT、T5、YOLOv8、Whisper# 、SAM等)也適用於所有後端。

Keras 3 高度向後相容 Keras 2:Keras 3 現在實作了 Keras 2 的公共 API 介面。大多數使用者無需更改任何程式碼即可在 Keras 3 上執行 Keras 腳本。如果你還不習慣使用 Keras 3,可以選擇忽略新版本的更新,繼續將 Keras 2 與 TensorFlow 結合使用。

Keras 3 支援所有後端的跨框架資料 pipeline。多框架機器學習也意味著多框架資料載入和預處理。 Keras 3 模型可以使用各種資料pipeline 進行訓練,無論你使用的是JAX、PyTorch 或TensorFlow 後端:

  • tf.data. Dataset pipelines。
  • torch.utils.data.DataLoader 物件。
  • NumPy 陣列和 Pandas 資料幀。
  • Keras 的 keras.utils.PyDataset 物件。

一個新的分散式 API,可用於大規模資料並行和模型並行#。目前這項更新僅適用於 JAX 後端,TensorFlow 和 PyTorch 支援即將推出。

至於為何要推出這項更改,Keras 團隊表示,近年來,隨著模型規模變得越來越大,他們希望為多設備模型分片( sharding)問題提供Keras 解決方案。這個團隊設計的 API 使模型定義、訓練邏輯和分片配置完全獨立,這意味著模型可以像在單一裝置上運行一樣, 然後,你可以在訓練模型時將分片配置新增到任意模型中。

資料並行(在多個裝置上相同地複製小模型)只需兩行即可處理:

Keras 3.0正式发布:可用于TensorFlow、JAX和PyTorch

接下來是模型並行。這個 API 允許你透過正規表示式配置每個變數和每個輸出張量的佈局。這使得為整個變數類別快速指定相同的佈局變得容易。

Keras 3.0正式发布:可用于TensorFlow、JAX和PyTorch

最後,Keras 團隊也收集了許多大家關心的問題,並予以解答,有興趣的讀者可以前去 Keras 官方網站,了解更多內容。

以上是Keras 3.0正式發布:可用於TensorFlow、JAX和PyTorch的詳細內容。更多資訊請關注PHP中文網其他相關文章!

本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn

熱門文章

倉庫:如何復興隊友
3 週前 By 尊渡假赌尊渡假赌尊渡假赌
Hello Kitty Island冒險:如何獲得巨型種子
3 週前 By 尊渡假赌尊渡假赌尊渡假赌
R.E.P.O.能量晶體解釋及其做什麼(黃色晶體)
1 週前 By 尊渡假赌尊渡假赌尊渡假赌

熱門文章

倉庫:如何復興隊友
3 週前 By 尊渡假赌尊渡假赌尊渡假赌
Hello Kitty Island冒險:如何獲得巨型種子
3 週前 By 尊渡假赌尊渡假赌尊渡假赌
R.E.P.O.能量晶體解釋及其做什麼(黃色晶體)
1 週前 By 尊渡假赌尊渡假赌尊渡假赌

熱門文章標籤

記事本++7.3.1

記事本++7.3.1

好用且免費的程式碼編輯器

SublimeText3漢化版

SublimeText3漢化版

中文版,非常好用

禪工作室 13.0.1

禪工作室 13.0.1

強大的PHP整合開發環境

Dreamweaver CS6

Dreamweaver CS6

視覺化網頁開發工具

SublimeText3 Mac版

SublimeText3 Mac版

神級程式碼編輯軟體(SublimeText3)

DeepMind機器人打乒乓球,正手、反手溜到飛起,全勝人類初學者 DeepMind機器人打乒乓球,正手、反手溜到飛起,全勝人類初學者 Aug 09, 2024 pm 04:01 PM

DeepMind機器人打乒乓球,正手、反手溜到飛起,全勝人類初學者

Claude也變懶了!網友:學會給自己放假了 Claude也變懶了!網友:學會給自己放假了 Sep 02, 2024 pm 01:56 PM

Claude也變懶了!網友:學會給自己放假了

首配機械爪!元蘿蔔亮相2024世界機器人大會,發布首個走進家庭的西洋棋機器人 首配機械爪!元蘿蔔亮相2024世界機器人大會,發布首個走進家庭的西洋棋機器人 Aug 21, 2024 pm 07:33 PM

首配機械爪!元蘿蔔亮相2024世界機器人大會,發布首個走進家庭的西洋棋機器人

世界機器人大會上,這家承載「未來養老希望」的國產機器人被包圍了 世界機器人大會上,這家承載「未來養老希望」的國產機器人被包圍了 Aug 22, 2024 pm 10:35 PM

世界機器人大會上,這家承載「未來養老希望」的國產機器人被包圍了

李飛飛團隊提出ReKep,讓機器人具備空間智能,還能整合GPT-4o 李飛飛團隊提出ReKep,讓機器人具備空間智能,還能整合GPT-4o Sep 03, 2024 pm 05:18 PM

李飛飛團隊提出ReKep,讓機器人具備空間智能,還能整合GPT-4o

鴻蒙智行享界S9全場景新品發表會,多款重磅新品齊發 鴻蒙智行享界S9全場景新品發表會,多款重磅新品齊發 Aug 08, 2024 am 07:02 AM

鴻蒙智行享界S9全場景新品發表會,多款重磅新品齊發

ACL 2024獎項發表:華科大破解甲骨文最佳論文之一、GloVe時間檢驗獎 ACL 2024獎項發表:華科大破解甲骨文最佳論文之一、GloVe時間檢驗獎 Aug 15, 2024 pm 04:37 PM

ACL 2024獎項發表:華科大破解甲骨文最佳論文之一、GloVe時間檢驗獎

國內首個UI大模型出爐! Motiff妙多大模型打造設計師最佳助手、優化UI設計工作流程 國內首個UI大模型出爐! Motiff妙多大模型打造設計師最佳助手、優化UI設計工作流程 Aug 19, 2024 pm 04:48 PM

國內首個UI大模型出爐! Motiff妙多大模型打造設計師最佳助手、優化UI設計工作流程

See all articles