I just tried to use Keras 3 with the script above and it seems to work (using tensorflow-metal 1.2.0). (replace: import tf_keras as keras with import keras)