KerasCore/Keras3.0
2023-11-28 本文已影响0人
unilearning
KerasCore是Keras3.0的预览版，其致力于在TF/Torch/Jax之上，构建统一的代码模块。在Keras3.0推出之后，KerasCore即更名为Keras3.0 。
原来的tf.keras代码可以无缝迁移到keras上，即将from tenrsorflow import keras更改为import keras_core as keras即可。
KerasCore其实是Keras的一种设计回归，在曾经初始阶段，Keras可以基于TF、CNTK、MXNET等多个后端，但是为了集中精力开发，后来所有的精力集中在Tensorflow的支持上。但是现在，Pytorch已经成为大家所喜爱的开发框架，Keras没有理由仅仅针对Tensorflow，因此Keras决定对于新的版本进行多后端的支持。
KerasCore具有以下特点：
- 全部API均适用于TF、Torch、Jax
- 利用keras.ops模块，访问底层框架的所有操作，实现一套代码适配多个框架
- 无缝整合进后端代码中，比如使用keras创建模型、使用torch创建optimizer组建训练流程
- 适配多个框架的数据处理流程
- 预训练模型支持：通过KerasCV、KerasNLP两个库，支持了众多预训练模型
- 提供高层API和底层API给用户，让用户完全按照自己的意愿开发
