Keras 社区发布 Kinetic 库,开发者通过装饰器即可将函数部署至云端 TPU/GPU 运行,定位类似 Modal 但新增 TPU 支持。该工具自动完成代码打包、Cloud Build 容器构建(支持缓存)、GKE 集群调度及结果返回,实现日志实时流式传输,使远程执行体验如同本地运行。
Perhaps the craziest thing that was introduced on the Keras community call today: Keras Kinetic, a new library that lets you run jobs on cloud TPU/GPU via a simple decorator -- like Modal but with TPU support.
When you call a decorated function, Kinetic handles the entire remote execution pipeline:
- Packages your function, local code, and data dependencies
- Builds a container with your dependencies via Cloud Build (cached after first build)
- Runs the job on a GKE cluster with the requested accelerator (TPU or GPU)
- Returns the result to your local machine (logs are streamed in real time, and the function's return value is delivered back as if it ran locally)