Since MLX is not supported on Mac Intel and AMD GPU, Multi-GPU support for Tensorflow-Metal is crucial.

For Tensorflow-metal > 1.0.0 Mac intel are not supported ! According to this table you must install Tensorflow==2.1.3 tensorflow tensorflow-metal MacOs features v2.5 v0.1.2 12.0+ Pluggable device v2.6 v0.2.0 12.0+ Variable seq. length RNN v2.7 v0.3.0 12.0+ Custom op support v2.8 v0.4.0 12.0+ RNN perf. improvements v2.9 v0.5.0 12.1+ Distributed training v2.10 v0.6.0 12.1+ v2.11 v0.7.0 12.1+ v2.12 v0.8.0 12.1+ v2.13 v1.0.0 12.1+ FP16 and BF16 support v2.14 v1.1.0 12.1+ Mac Silicon only

Work fine for me, with with this function "layers.experimental.preprocessing": tf.device('/cpu:0'): data_augmentation = keras.Sequential( [ layers.experimental.preprocessing.RandomFlip("horizontal"), layers.experimental.preprocessing.RandomRotation(0.05), ] )

Yes ! My setup : Mac mini 8,3 + 5700XT + AKITIO Node eGPU Box TensorFlow Metal 2.6