Tested @Frameworks Engineer's method. For the example in https://developer.apple.com/metal/tensorflow-plugin/, just adding from tensorflow.keras.optimizers.legacy import Adam and modifying 'adam' to Adam() works for me.