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.