À propos du projet
Keras 3 est un framework d'apprentissage profond de haut niveau conçu pour la flexibilité et la performance sur plusieurs backends. Il prend en charge JAX, TensorFlow et PyTorch pour l'entraînement et l'inférence, avec un support supplémentaire limité à l'inférence pour OpenVINO.
Les capacités clés incluent :
- Support Multi-Backend : Les utilisateurs peuvent basculer entre JAX, TensorFlow et PyTorch pour tirer parti des avantages spécifiques de chaque framework, tels que la performance de JAX ou l'écosystème de production de TensorFlow.
- Large Gamme d'Applications : Prend en charge le développement de modèles pour la vision par ordinateur, le traitement du langage naturel, le traitement audio, la prévision de séries temporelles et les systèmes de recommandation.
- Interopérabilité : Les modèles Keras peuvent être intégrés dans des Modules PyTorch natifs ou des fonctions de modèle natives JAX. Il prend également en charge des jeux de données de divers formats, y compris tf.data.Dataset et les PyTorch DataLoaders.
- Évolutivité : Conçu pour passer d'ordinateurs portables locaux à de grands clusters de GPU ou TPU.
- Compatibilité Rétroactive : Agit comme un remplacement direct pour tf.keras lors de l'utilisation du backend TensorFlow.