多后端深度学习框架,一份代码跑 TensorFlow/JAX/PyTorch
Star64k
24h 新增+1
7 天新增—
Fork19k
Issue285
语言Python
项目介绍
更新于 1 天前
Keras 是一个面向深度学习的高级框架,主打「Deep Learning for humans」,用简洁的 API 帮你快速搭建和训练模型,覆盖计算机视觉、自然语言处理、时序预测、推荐系统等常见任务。Keras 3 最大的变化是变成多后端框架,同一份模型代码可以跑在 TensorFlow、JAX、PyTorch 上,推理还能切到 OpenVINO,相当于在 PyTorch 和 TensorFlow 之间不再二选一,还能按模型结构挑最快的后端拿性能收益。它兼容原有 tf.keras 代码,也能直接消费 tf.data 或 PyTorch 的 DataLoader,方便避免框架锁定。适合需要快速原型、跨框架迁移或在 JAX/PyTorch 上补一个成熟训练库的算法工程师和 ML 研究者;如果你只做纯推理部署或不写模型代码,关系不大。
上手成本:中
Star 趋势
话题标签
data-sciencedeep-learningjaxmachine-learningneural-networkspythonpytorchtensorflow
仓库信息
| License | Apache-2.0 |
| 创建时间 | 2015-03-28 |
| 最后推送 | 5 小时前 |
| 热度分 | 8.0 |