I'm following the instruction here: https://developer.apple.com/metal/tensorflow-plugin/ and it works