Update: It is finally running on my M1. Please install the release from apple/tensorflow_macros. Also, make sure that you run the script in the right arch i.e. arm64
How to install TensorFlow and NumPy on Apple Silicon (M1)