动态

Keras Kinetic:通过装饰器在云TPU/GPU运行作业

Keras Kinetic:通过装饰器在云TPU/GPU运行作业
François Chollet
今天的 Keras 社区电话会议上介绍的最疯狂的东西可能是:Keras Kinetic,一个新的库,让你通过一个简单的装饰器在云 TPU/GPU 上运行作业——像 Modal 但支持 TPU。

当你调用一个装饰过的函数时,Kinetic 处理整个远程执行流程:

- 打包你的函数、本地代码和数据依赖
- 通过 Cloud Build 构建一个包含你的依赖的容器(首次构建后缓存)
- 在带有请求的加速器(TPU 或 GPU)的 GKE 集群上运行作业
- 将结果返回给你的本地机器(日志实时流式传输,函数的返回值像本地运行一样传递回来)
动态François Chollet2026-04-03原文

相关内容