Vue d'ensemble

TensorFlow 2.17, publié le 7 août 2024, apporte des améliorations de performance pour l'entraînement et l'inférence des modèles.

Fonctionnalités principales

Performances améliorées

Les optimisations du compilateur XLA et du runtime permettent des gains de performance significatifs sur les workloads d'entraînement et d'inférence.

python 
import tensorflow as tf

# Modèle simple avec compilation XLA
model = tf.keras.Sequential([
    tf.keras.layers.Dense(128, activation='relu'),
    tf.keras.layers.Dense(10, activation='softmax'),
])

model.compile(
    optimizer='adam',
    loss='sparse_categorical_crossentropy',
    jit_compile=True,  # compilation XLA
)
print(f'TensorFlow {tf.__version__}')

Sources