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给用户,让用户完全按照自己的意愿开发
网友评论