keras-team/keras

Deep Learning for humans

What it solves

Keras 3 是一个高级深度学习框架，旨在消除框架锁定。它允许开发者为各种任务（如计算机视觉、自然语言处理和音频处理）构建和训练模型，同时对实际执行计算的底层引擎保持中立。

How it works

Keras 3 充当多后端接口。它提供一致的高级 API 用于模型开发，随后可以在 JAX、TensorFlow、PyTorch 或 OpenVINO（仅用于推理）之上运行。用户只需配置环境变量即可在这些后端之间切换，让他们在不重写代码的情况下，利用各框架的性能或生态系统优势。

Who it’s for

它面向深度学习实践者，从个人开发者到全球企业，都希望加速模型开发并避免被单一后端框架绑定。

Highlights

  • Multi-backend support: Compatible with JAX, TensorFlow, and PyTorch for training, and OpenVINO for inference.
  • Performance optimization: Ability to pick the fastest backend for a specific architecture, potentially increasing speed by 20% to 350%.
  • Interoperability: Keras models can use datasets from any format, including tf.data.Dataset and PyTorch DataLoaders.
  • wysokie scalability: Scales from local laptops to large GPU or TPU clusters.