kale

GitHub
683 156 困难 1 次阅读 2天前Apache-2.0开发框架数据工具
AI 解读 由 AI 自动生成,仅供参考

Kale(Kubeflow Automated pipeLines Engine)是专为数据科学家打造的 Kubeflow 增强工具,旨在简化机器学习工作流的部署过程。它就像数据科学领域的“超级食物”,让复杂的流程编排变得轻松可口。

在 Kubernetes 上运行 Kubeflow 虽然功能强大,但直接使用其 SDK 编写和维护流水线对许多专注于模型探索的研究人员来说门槛较高。Kale 完美解决了这一痛点:它允许用户直接在熟悉的 JupyterLab 界面中,通过简单的图形化操作将笔记本代码转化为可复用的 Kubeflow 流水线,全程无需修改任何一行代码。这不仅保留了数据探索的交互性,还实现了从实验到生产部署的无缝衔接。

Kale 特别适合需要在交互式环境中进行迭代建模的数据科学家和研究人员,尤其是那些希望利用 Kubernetes 的强大算力,却不希望深陷复杂编排系统细节的用户。其核心亮点在于“零代码转换”能力,自动识别笔记本中的代码块并将其封装为流水线步骤,极大地降低了分布式任务 orchestration 的技术壁垒。目前项目已重启开发并致力于全面兼容最新的 Kubeflow Pipelines v2 版本,为社区带来更现代化的自动化体验。

使用场景

某电商公司的数据科学团队正在基于 Jupyter Notebook 迭代开发一个用户流失预测模型,并需要将其部署为可重复执行的自动化流程。

没有 kale 时

  • 代码重构负担重:数据科学家必须手动将探索性的 Notebook 代码拆解为独立的 Python 脚本,并重新编写大量样板代码以适配 Kubeflow Pipelines SDK。
  • 运维门槛高:团队成员需深入学习 Kubernetes 配置和管道编排逻辑,非专家往往因环境配置错误导致部署失败。
  • 迭代效率低下:每次调整模型参数或特征工程逻辑后,都需重复“修改代码 - 重新打包 - 上传管道”的繁琐过程,严重拖慢实验节奏。
  • 协作断层明显:交互式探索环境与生产级管道之间存在巨大鸿沟,导致模型从实验到上线的周期长达数周。

使用 kale 后

  • 零代码迁移:直接在 JupyterLab 界面中通过注释标记代码块,kale 自动将 Notebook 转换为标准的 Kubeflow 管道,无需改动原有逻辑。
  • 自助式部署:提供直观的 UI 配置资源限额和依赖关系,数据科学家无需精通 K8s 即可一键提交复杂的工作流。
  • 敏捷迭代闭环:修改 Notebook 后立即重新生成管道版本,将原本数小时的部署工作缩短至分钟级,大幅提升实验频率。
  • 无缝衔接生产:消除了探索环境与生产环境的差异,确保模型在开发阶段的行为与线上运行完全一致,加速价值交付。

kale 的核心价值在于让数据科学家专注于算法创新,而非被复杂的工程化部署流程所束缚。

运行环境要求

操作系统
  • Linux
  • macOS
  • Windows
GPU

非必需(步骤单元格可自定义 GPU 需求)

内存

未说明

依赖
notes需要 Kubernetes 集群(如 minikube, kind)和 Kubeflow Pipelines v2.16.0+ 环境。Kale v2.0 尚未发布到 PyPI,目前需从源码安装。所有导入语句必须放在标记为'imports'的单元格中,否则会导致运行时错误。支持通过 Docker 进行本地测试。
python3.11+
kubeflow-kale
jupyterlab>=4.0.0
kfp[kubernetes]>=2.16.0
kfp>=2.0.0
kale hero image

快速开始

Kale Logo

GitHub License PyPI Version npm Version Kale CI Workflow Status


[!NOTE]

项目状态更新 🚀

经过数年暂停后,我们很高兴地宣布 Kale 的开发已重新启动!🎉 当年,Kale 曾广受社区好评。如今,我们的目标是通过将所有组件更新至最新版本,并确保与最新 Kubeflow 发布版本完全兼容,重新建立一个稳固的基础。

更多详情请参阅 迈向 2.0 版本的议题

KALE(Kubeflow 自动化流水线引擎)是一个旨在简化数据科学人员部署 Kubeflow Pipelines 工作流体验的项目。 Kubeflow 是一个优秀的平台,用于在 Kubernetes 上编排复杂的流程;而 Kubeflow Pipelines 则提供了创建可重用组件的方法,这些组件可以作为工作流的一部分执行。Kubeflow 的自助服务特性使其对数据科学领域极具吸引力,因为它能够轻松访问高级分布式作业编排、组件复用、Jupyter Notebook、丰富的用户界面等功能。然而,对于可能不熟悉编排平台及相关 SDK 的数据科学家来说,开发和维护 Kubeflow 工作流仍然颇具挑战性。此外,数据科学通常涉及数据探索、迭代建模以及交互式环境(主要是 Jupyter Notebook)等环节。 Kale 通过提供一个简单的 UI,使用户可以直接从 JupyterLab 界面定义 Kubeflow Pipelines 工作流,而无需修改任何代码行,从而弥合了这一差距。

更多详细信息,请参阅 README 底部的 Kale v2.0 演示 视频。 有关 Kale 及其工作原理的更多信息,请阅读这篇 Medium 文章: 使用 Kale 将 Jupyter Notebook 部署自动化到 Kubeflow Pipelines

开始使用

要求

  • Python 3.11+
  • Kubeflow Pipelines v2.16.0+
    • 较旧的 KFP 服务器(kfp[kubernetes] < 2.16.0)无法识别 Kubernetes 执行器配置中的 securityContext 字段,这会导致流水线提交失败。
    • 按照官方 Kubeflow Pipelines 安装文档 中的建议安装 KFP(请确保设置 PIPELINE_VERSION=2.16.0 或更高版本)。
    • 如果您是从早期版本升级而来,请确保您的依赖项中包含 kfp[kubernetes]>=2.16.0kfp>=2.0.0
  • 一个 Kubernetes 集群(minikubekind 或任何 K8s 集群)

安装

[!IMPORTANT] Kale v2.0 尚未发布到 PyPI。 在此之前,请从源代码安装:

git clone https://github.com/kubeflow-kale/kale.git
cd kale
make dev      # 设置开发环境
make jupyter  # 启动 JupyterLab

详细的设置说明请参阅 CONTRIBUTING.md。 待 v2.0 发布后,您即可通过 PyPI 安装:

pip install "jupyterlab>=4.0.0" kubeflow-kale[jupyter]
jupyter lab

验证安装

  1. 启动您的 Kubernetes 集群和 KFP:

    minikube start
    kubectl port-forward -n kubeflow svc/ml-pipeline-ui 8080:80
    
  2. 测试命令行工具:

    kale --nb examples/base/candies_sharing.ipynb --kfp_host http://127.0.0.1:8080 --run_pipeline
    

    这将生成一个 .kale/ 目录下的流水线文件,并将其提交到 KFP。

  3. 测试 JupyterLab 扩展:

    • 打开 JupyterLab(运行 make jupyterjupyter lab
    • 打开 examples/base/ 中的一个笔记本
    • 点击左侧面板中的 Kale 图标
    • 使用切换按钮启用 Kale 面板
Kale JupyterLab 扩展

Docker(本地测试)

您可以通过 Docker 在类似 Kubeflow 的 Notebook 环境中测试 Kale。该镜像基于官方的 Kubeflow Notebook 镜像(jupyter-scipy),并预先安装了 Kale。

make docker-build   # 构建 wheel 文件 + Docker 镜像
make docker-run     # 在 http://localhost:8889 启动 JupyterLab

要连接到 KFP 集群,请在不同的终端中分别运行以下命令:

# 终端 1:提供开发版 wheel 文件服务(以便编译后的流水线可以安装 Kale)
make kfp-serve

# 终端 2:转发 KFP API 端口
kubectl port-forward -n kubeflow svc/ml-pipeline 8080:8888

# 终端 3:启动容器
make docker-run

make docker-run 会自动配置:

  • KFP API 通过 host.docker.internal 访问(适用于 macOS、Windows 和 Linux)
  • KFP UI 链接 指向 localhost:8080(以便流水线链接能在浏览器中打开)
  • Wheel 服务器 连通性,方便编译后的流水线安装

单元格类型

Kale 使用特殊的单元格类型(标签)来将您的笔记本组织成流水线组件。您可以使用 Kale 的 JupyterLab 扩展,或直接在笔记本元数据中添加标签,为单元格分配这些类型。

细胞类型参考

细胞类型 状态 描述
导入 ✅ 正常工作 该细胞中的代码将在流水线的每一步之前插入。用于所有导入语句。所有导入必须放在标记为 imports 的细胞中。 在其他类型的细胞中导入库(pandas、tensorflow 等)会导致流水线执行错误。
函数 ✅ 正常工作 该细胞中的代码将在 imports 之后,插入到流水线的每一步之前。仅用于函数和类的定义。请勿包含 顶级可执行语句
流水线参数 ✅ 正常工作 定义将成为流水线参数的变量。如果存在多个流水线参数细胞,并且每个细胞都定义了一个参数,则仅采用最终值。
流水线指标 ✅ 正常工作 打印标量指标并将其转换为流水线指标。
步骤 ✅ 正常工作 具有自定义名称的常规流水线步骤。这是您数据处理和机器学习逻辑的默认细胞类型。每个步骤可以依赖于其他步骤。步骤也可以定义自己的镜像和 GPU 要求。
跳过细胞 ✅ 正常工作 标记为跳过的细胞将被排除在流水线之外。适用于探索性代码或调试,这些内容不应成为生产流水线的一部分。

重要指南

[!WARNING] 不在 Imports 细胞中的导入将不会被检测以进行自动依赖项安装,这会导致运行时出现 ImportError,如果容器镜像中未预先安装该包。

最佳实践:

  • 将所有导入放在笔记本的开头,并置于标记为 Imports 的细胞中
  • 保持函数定义的纯粹性——无副作用(修改全局变量或可变参数)、打印或导入
  • 使用 pipeline-parameters 来存储您可能希望在不同运行之间调整的值
  • 使用 skip 细胞来进行不应包含在流水线中的探索性分析

示例

请查看 examples/ 中的示例笔记本,以了解细胞类型的实际应用。

常见问题解答

请前往 FAQ 阅读一些已知问题以及 Kale 数据编排模型所施加的一些限制。

资源

贡献

make dev      # 设置开发环境
make test     # 运行所有测试
make jupyter  # 启动 JupyterLab

请参阅 CONTRIBUTING.md,获取详细的开发说明,包括:

  • 可用的 make 命令
  • 使用 KFP 集群进行测试
  • 构建发布工件
  • 实时重载设置

Kale v2.0 演示

观看下方 Kubeflow Kale 演示——入门视频。

演示

版本历史

v2.0.0rc12026/04/02
v2.0.0a52026/03/19
v2.0.0a42026/03/12
v0.7.02021/05/19
v0.6.12021/01/28
v0.6.02020/12/10
v0.5.12020/09/11
v0.5.02020/06/15
v0.4.02020/04/14
v.0.1.22019/05/13

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|2周前
Agent开发框架图像

n8n

n8n 是一款面向技术团队的公平代码(fair-code)工作流自动化平台,旨在让用户在享受低代码快速构建便利的同时,保留编写自定义代码的灵活性。它主要解决了传统自动化工具要么过于封闭难以扩展、要么完全依赖手写代码效率低下的痛点,帮助用户轻松连接 400 多种应用与服务,实现复杂业务流程的自动化。 n8n 特别适合开发者、工程师以及具备一定技术背景的业务人员使用。其核心亮点在于“按需编码”:既可以通过直观的可视化界面拖拽节点搭建流程,也能随时插入 JavaScript 或 Python 代码、调用 npm 包来处理复杂逻辑。此外,n8n 原生集成了基于 LangChain 的 AI 能力,支持用户利用自有数据和模型构建智能体工作流。在部署方面,n8n 提供极高的自由度,支持完全自托管以保障数据隐私和控制权,也提供云端服务选项。凭借活跃的社区生态和数百个现成模板,n8n 让构建强大且可控的自动化系统变得简单高效。

184.7k|★★☆☆☆|今天
数据工具开发框架Agent

AutoGPT

AutoGPT 是一个旨在让每个人都能轻松使用和构建 AI 的强大平台,核心功能是帮助用户创建、部署和管理能够自动执行复杂任务的连续型 AI 智能体。它解决了传统 AI 应用中需要频繁人工干预、难以自动化长流程工作的痛点,让用户只需设定目标,AI 即可自主规划步骤、调用工具并持续运行直至完成任务。 无论是开发者、研究人员,还是希望提升工作效率的普通用户,都能从 AutoGPT 中受益。开发者可利用其低代码界面快速定制专属智能体;研究人员能基于开源架构探索多智能体协作机制;而非技术背景用户也可直接选用预置的智能体模板,立即投入实际工作场景。 AutoGPT 的技术亮点在于其模块化“积木式”工作流设计——用户通过连接功能块即可构建复杂逻辑,每个块负责单一动作,灵活且易于调试。同时,平台支持本地自托管与云端部署两种模式,兼顾数据隐私与使用便捷性。配合完善的文档和一键安装脚本,即使是初次接触的用户也能在几分钟内启动自己的第一个 AI 智能体。AutoGPT 正致力于降低 AI 应用门槛,让人人都能成为 AI 的创造者与受益者。

183.6k|★★★☆☆|今天
Agent语言模型插件

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|2周前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

161.1k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

109.2k|★★☆☆☆|2天前
开发框架图像Agent