多后端支持
官网指南分别给出在 JAX、TensorFlow 与 PyTorch 下自定义 fit() 与自定义训练循环的做法,同一套 API 可以切换后端。
拖动滑块完成拼图,通过后发送验证码
独立第三方 AI 评测平台 · 只用实测数据
Keras 是 keras.io 上的深度学习 API,它面向人类开发者而非机器,主打调试速度、代码简洁、可维护与可部署,多后端方式让它能跑在 JAX、TensorFlow 与 PyTorch 上。的开发者指南多为 Jupyter notebook,可在 Google Colab 一键运行,Colab 带 GPU 与 TPU 运行时;指南覆盖函数式 API、子类化、自定义训练循环、分布式训练、迁移学习与量化等主题。短板是:包括的三个页面里没有价格页与商业支持信息,公开介绍把它标为免费与开源项目;函数式 API 指南页头显示最后修改日期为 2023/06/25,文档是否持续更新要自行核对。
Keras 对应 keras.io,是一个深度学习 API,官网首页把它定义为面向人类开发者而不是机器的接口,关注点是调试速度、代码优雅与简洁、可维护性与可部署性;选用它之后代码库更小、更易读、更容易迭代。导航分为 API DOCS、GUIDES 与 EXAMPLES 三块。
按官网开发者指南页的说明,指南是对具体主题的深入讲解,例如层子类化、微调与模型保存,多数以 Jupyter notebook 形式发布,可在 Google Colab 一键运行,不需要本地配置;Colab 提供 GPU 与 TPU 运行时。指南清单里可以看到函数式 API、Sequential 模型、子类化建层、内置训练与评估方法、在 JAX 与 TensorFlow 与 PyTorch 下自定义 fit()、自定义训练循环、序列化与保存、回调、迁移学习与微调、分布式训练、Keras 2 迁移到 Keras 3、量化(含 8-bit 与 4-bit 整数量化、GPTQ、AWQ)、导出到 LiteRT 以及自定义 TPU/GPU Kernel 等主题。
涉及费用与商用条款需要另查官网。
官网指南分别给出在 JAX、TensorFlow 与 PyTorch 下自定义 fit() 与自定义训练循环的做法,同一套 API 可以切换后端。
把模型看作图层组成的有向无环图,可处理非线性拓扑、共享层以及多输入多输出,比 Sequential 更灵活。
函数式 API 指南 ↗Sequential 模型是官网上与函数式 API 并列的基础建模范式,适合层与层顺序堆叠的简单结构。
官网指南讲层与模型子类化,适合需要自定义前向逻辑或非标准结构的场景,代价是代码量更大。
Model 自带 fit() 训练循环与 evaluate() 评估循环,也可按指南改写成自定义训练步骤,训练、评估与推理用法一致。
用 model.save() 把整个模型存成单个文件,之后再 load_model() 还原,文件含架构、权重、训练配置与优化器状态。
有迁移学习与微调专题指南,也覆盖训练与评估、自定义回调等常规流程环节。
指南覆盖 8-bit 与 4-bit 整数量化、GPTQ 与 AWQ、量化兼容层,以及把 Keras 模型导出到 LiteRT。
当你需要一套写起来短、切后端方便、能覆盖从建模到部署的深度学习 API 时,Keras 值得用:它面向人类开发者,多后端可跑在 JAX、TensorFlow 与 PyTorch 上,指南多为 notebook 且能在 Google Colab 一键运行,函数式 API 示例在 MNIST 上两个 epoch 就到 0.9613 的测试准确率。要注意三点:费用与条款要另查;LiteRT 导出这类能力以指南文档为主,落地前需自行验证;版本跨度较大时,Keras 2 到 Keras 3 的迁移要留出改造时间。
想要代码短、易读、好调试的建模体验,函数式 API 与内置训练循环能覆盖大部分日常模型开发。
需要在 JAX、TensorFlow 与 PyTorch 之间切换或共用一套上层 API 时,多后端方式能省掉重复实现。
指南以 notebook 形式发布并可在 Google Colab 一键运行,不需要本地环境就能跟着示例上手。
页面没有价格、授权与支持条款,走采购流程时要另外确认服务与责任主体。
提供的是 API 文档与代码指南,没有可视化建模界面。
| 指标 | 数值 | 数据来源 |
|---|---|---|
| MNIST 示例测试准确率 | 0.9613 | The Functional API 指南运行输出 |
| MNIST 训练样本数 | 60000 | The Functional API 指南示例代码 |
| MNIST 测试样本数 | 10000 | The Functional API 指南示例代码 |
| 示例批量大小 | batch_size = 64 | The Functional API 指南示例代码 |
| 多输入示例标签数 | num_tags = 12 | The Functional API 指南示例代码 |
| 多输入示例词表大小 | num_words = 10000 | The Functional API 指南示例代码 |
| 多输入示例部门数 | num_departments = 4 | The Functional API 指南示例代码 |
| toy ResNet 演示样本量 | 前 1000 个样本 | The Functional API 指南示例代码 |
| 量化支持位宽 | 8-bit 与 4-bit 整数量化 | Developer guides 清单 |
| 函数式 API 指南最后修改 | 2023/06/25 | The Functional API 指南页头 |
包括的三个页面里没有定价页,也没有出现价格数字,所以这里不写具体费率。公开介绍把它标为免费与开源项目,官网导航只有 API DOCS、GUIDES 与 EXAMPLES。实际费用与商用条款请以官网为准,需要采购背书时建议直接确认授权说明。
它是面向人类开发者的深度学习 API,关注调试速度、代码简洁、可维护与可部署。多后端方式让它能跑在 JAX、TensorFlow 与 PyTorch 上,指南覆盖函数式 API、子类化、自定义训练循环、分布式训练、迁移学习、量化与模型导出。
函数式 API 比 keras.Sequential 更灵活,能处理非线性拓扑、共享层以及多输入多输出;Sequential 适合层按顺序堆叠的简单结构。指南里的多输入示例同时输入标题、正文与标签,输出优先级与部门两个结果,这种结构只能用函数式 API 表达。
官网把 JAX、TensorFlow 与 PyTorch 当作可切换的后端,并分别提供在这些框架下自定义 fit() 与训练循环、以及分布式训练的指南。如果团队已有大量 PyTorch 代码,可以继续用原框架,或把 Keras 当作上层 API;需要一套代码跨后端复用时,多后端方式更省事。
不一定。多数指南是 Jupyter notebook,可在 Google Colab 一键运行,不需要本地配置,Colab 自带 GPU 与 TPU 运行时。想本地跑就按指南里的 import 片段准备 numpy 与 keras,示例代码可以直接复制执行。
示例展示:先加载数据并 reshape,再 compile 指定损失、优化器与指标,然后 fit 训练并 evaluate 评估;保存时调用 model.save() 存成单个文件,之后用 load_model() 还原。该文件包含模型架构、权重、训练配置以及优化器及其状态,代码不在也能重建模型。
官网页面没有提到地区限制或访问要求,也没有价格与支持条款。需要注意的是指南默认用 Google Colab 运行,国内访问 Colab 可能不稳定,可改为本地环境跑同一份 notebook;量化与导出等部署环节以文档说明为准,落地前建议自行验证。
| 工具 | 编辑评分 | 价格 | 核心功能 | 平台 |
|---|---|---|---|---|
| Keras本工具 | 4.8 | 免费 | 多后端支持;函数式 API;序列式模型;子类化建模 | Web / Python / Google Colab |
| PyTorch | — | 未标价 | PyTorch 是一个开源深度学习框架,提供了丰富的工具和库,支持计算机视觉、自然语言处理 | Web / Linux / Mac |
| MLX | — | 未标价 | MLX 是一个用于机器学习的库,提供了丰富的数组操作、函数变换、随机数生成等功能 | — |
| TensorFlow | — | 未标价 | tensorflow 官网,google 推出的机器学习和人工智能开源库,gpu,pytho | — |
| Lobe | — | 为 free,即免费使用 | Lobe 是一款免费的桌面应用程序,由相关团队开发,旨在让机器学习变得简单易用 | Mac / PC |
| DL4J | — | 未标价 | Deeplearning4j 是一套用于在 JVM 上运行深度学习的工具套件 | — |
指南页头标注 Date created: 2019/03/01,Last modified: 2023/06/25,内容覆盖函数式 API 的建模、训练、保存与复杂拓扑写法。
官方公告 ↗