Support TensorFlow >= 2.16 and Keras 3