Redian新闻
>
Keras 3.0正式发布:可用于TensorFlow、JAX和PyTorch

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

公众号新闻

机器之心报道

编辑:陈萍

经过 5 个月的更新迭代,Keras 3.0 终于来了。

「大新闻:我们刚刚发布了 Keras 3.0 版本!」Keras 之父 François Chollet 在 X 上激动的表示。「现在你可以在 JAX、TensorFlow 以及 PyTorch 框架上运行 Keras……」


对于这一更新,Keras 官方表示,这一版本足足花了他们 5 个月的时间进行公测才完成。Keras 3.0 是对 Keras 的完全重写,你可以在 JAX、TensorFlow 或 PyTorch 之上运行 Keras 工作流,新版本还具有全新的大模型训练和部署功能。你可以选择最适合自己的框架,也可以根据当前的目标从一种框架切换到另一种框架都没有问题。


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

被 250 多万开发者使用的 Keras,迎来 3.0 版本

Keras API 可用于 JAX、TensorFlow 和 PyTorch。现有的仅使用内置层的 tf.keras 模型可以在 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 中使用各种预训练模型。现在已经有 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 使模型定义、训练逻辑和分片配置完全独立,这意味着模型可以像在单个设备上运行一样, 然后,你可以在训练模型时将分片配置添加到任意模型中。

数据并行(在多个设备上相同地复制小模型)只需两行即可处理:


接下来是模型并行。该 API 允许你通过正则表达式配置每个变量和每个输出张量的布局。这使得为整个变量类别快速指定相同的布局变得容易。


最后,Keras 团队还收集了很多大家关心的问题,并予以解答,感兴趣的读者可以前去 Keras 官方网站,了解更多内容。





© THE END 

转载请联系本公众号获得授权

微信扫码关注该文公众号作者

戳这里提交新闻线索和高质量文章给我们。
相关阅读
NeurIPS 2023 | MixFormerV2:基于Transformer的高效跟踪器【玩具日报】名创优品涨近7% 全球首家盲盒店在伦敦开业;泡泡玛特旗下首家实体画廊inner flow Gallery正式揭幕人事五则|Dior任命传讯总监,Guess联合创始人退休;Salomon、Interparfums、Jumeirah等管理层变动Yarn 4.0正式发布,现代化的软件包管理器中国版Copilot?代码优化提效5倍、采纳率提升44%……讯飞iFlyCode2.0正式发布,各项能力全面开放精选SDE岗位 | Siemens、Applied Materials、Jane Street公司岗位发布!掌握Wind, Excel和Python三大金融必备技能, 实现年薪翻两番一篇文章带你了解Python常用自动化测试框架——PytestPython 3.12 正式发布:性能提升、no-GIL将在 3.13 提供夏婳:两情难相知(十七)少林方丈释永信谈AI与禅;天玑9300正式发布;黄仁勋希望由成龙扮演自己PyTorch大更新!编译代码速度暴增35倍!视觉模型一键部署,头显Quest 3可用早鸟报|文心大模型4.0正式发布;华为回应“花瓣支付”更名;东方甄选正式宣布推出付费会员业务...AI,美丽新世界?养虎为患?[评测]MSI MPG GUNGNIR 300P AIRFLOW 评测Python实战 | 使用 Python 和 TensorFlow 构建卷积神经网络(CNN)进行人脸识别ChatGPT重压下,Stack Overflow裁员28%,为自家生成式AI工具开源节流首销1399元起,vivo Y100正式发布,高颜值长续航Keras 3.0一统江湖!大更新整合PyTorch、JAX,全球250万开发者在用了BB鸭 | vivo X100正式发布;小红书注册“小地瓜”商标;小米汽车信息首曝;红魔9 Pro官宣11月必看!“新世界三大男高音”Juan Diego Flórez首次亮相澳洲![电脑] 立式无光ITX小钢炮,ROG B760-I+ProArt 4070 Ti+海盗船 2000D AIRFLOW 装机分享百度文心大模型4.0正式发布并开启邀测;YouTube推出广告套餐,利用AI投放广告丨AIGC日报大嫂杨绒花(下)专访丨积家CEO Catherine Rénier:情感联结和艺术表达对腕表也至关重要PyTorch大更新,编译代码速度暴增35倍!视觉模型一键部署,头显Quest 3可用PyTorch2.1支持昇腾NPU!华为成为中国首个PyTorch基金会Premier会员New Dam in NW China Aims to Restore Flow to Parched Heihe River基于Transformer和注意力的可解释核苷酸语言模型,用于pegRNA优化设计【已生效】万豪套房券改名为升房券:可用于丽思卡尔顿、艾迪逊等品牌,确认时间缩短到3天【天与湖】 一组小诗 6Keras 3.0正式发布!一统TF/PyTorch/Jax三大后端框架,网友:改变游戏规则OpenHarmony 4.0正式发布,增删改代码8849882行现代文明的思维逻辑价值判断PyTorch最新工具mm来了!3D可视化矩阵乘法、Transformer注意力!
logo
联系我们隐私协议©2024 redian.news
Redian新闻
Redian.news刊载任何文章,不代表同意其说法或描述,仅为提供更多信息,也不构成任何建议。文章信息的合法性及真实性由其作者负责,与Redian.news及其运营公司无关。欢迎投稿,如发现稿件侵权,或作者不愿在本网发表文章,请版权拥有者通知本网处理。